<!DOCTYPE html PUBLIC "-//W3C//DTD XHTML 1.0 Transitional//EN" "https://www.w3.org/TR/xhtml1/DTD/xhtml1-transitional.dtd">
<html xmlns="http://www.w3.org/1999/xhtml" lang="en-US">
<head>
<meta http-equiv="Content-Type" content="text/xhtml;charset=UTF-8"/>
<meta http-equiv="X-UA-Compatible" content="IE=11"/>
<meta name="generator" content="Doxygen 1.12.0"/>
<meta name="viewport" content="width=device-width, initial-scale=1"/>
<title>NeuZephyr: D:/Users/Mgepahmge/Documents/C Program/NeuZephyr/src/ComputeGraph.cu Source File</title>
<link rel="icon" href="NZ_logo2.png" type="image/x-icon" />
<link href="tabs.css" rel="stylesheet" type="text/css"/>
<script type="text/javascript" src="jquery.js"></script>
<script type="text/javascript" src="dynsections.js"></script>
<link href="navtree.css" rel="stylesheet" type="text/css"/>
<script type="text/javascript" src="resize.js"></script>
<link href="doxygen.css" rel="stylesheet" type="text/css" />
</head>
<body>
<div id="top"><!-- do not remove this div, it is closed by doxygen! -->
<div id="titlearea">
<table cellspacing="0" cellpadding="0">
 <tbody>
 <tr id="projectrow">
  <td id="projectlogo"><img alt="Logo" src="NZ_logo2.png"/></td>
  <td id="projectalign">
   <div id="projectname">NeuZephyr
   </div>
   <div id="projectbrief">Simple DL Framework</div>
  </td>
 </tr>
 </tbody>
</table>
</div>
<!-- end header part -->
<!-- Generated by Doxygen 1.12.0 -->
<script type="text/javascript">
/* @license magnet:?xt=urn:btih:d3d9a9a6595521f9666a5e94cc830dab83b65699&amp;dn=expat.txt MIT */
$(function() { codefold.init(0); });
/* @license-end */
</script>
  <div id="navrow1" class="tabs">
    <ul class="tablist">
      <li><a href="index.html"><span>Main&#160;Page</span></a></li>
      <li><a href="pages.html"><span>Related&#160;Pages</span></a></li>
      <li><a href="namespaces.html"><span>Namespaces</span></a></li>
      <li><a href="annotated.html"><span>Classes</span></a></li>
      <li class="current"><a href="files.html"><span>Files</span></a></li>
    </ul>
  </div>
  <div id="navrow2" class="tabs2">
    <ul class="tablist">
      <li><a href="files.html"><span>File&#160;List</span></a></li>
    </ul>
  </div>
<script type="text/javascript">
/* @license magnet:?xt=urn:btih:d3d9a9a6595521f9666a5e94cc830dab83b65699&amp;dn=expat.txt MIT */
$(function(){ initResizable(false); });
/* @license-end */
</script>
<div id="nav-path" class="navpath">
  <ul>
<li class="navelem"><a class="el" href="dir_d522931ffa1371640980b621734a4381.html">Users</a></li><li class="navelem"><a class="el" href="dir_a7e6ee1ae3f772c9504a0b543f2027e2.html">Mgepahmge</a></li><li class="navelem"><a class="el" href="dir_e03f57e346cc4845a4c354a35630b169.html">Documents</a></li><li class="navelem"><a class="el" href="dir_231a0482af2b83c895f27ba7fe745141.html">C Program</a></li><li class="navelem"><a class="el" href="dir_0fa7fc3a0dfd304dbfc9dce9f6facfa2.html">NeuZephyr</a></li><li class="navelem"><a class="el" href="dir_25794e61537e3f33113e2168c9f8da60.html">src</a></li>  </ul>
</div>
</div><!-- top -->
<div id="doc-content">
<div class="header">
  <div class="headertitle"><div class="title">ComputeGraph.cu</div></div>
</div><!--header-->
<div class="contents">
<div class="fragment"><div class="line"><a id="l00001" name="l00001"></a><span class="lineno">    1</span><span class="preprocessor">#include &quot;<a class="code" href="_compute_graph_8cuh.html">NeuZephyr/ComputeGraph.cuh</a>&quot;</span></div>
<div class="line"><a id="l00002" name="l00002"></a><span class="lineno">    2</span><span class="preprocessor">#include &quot;NeuZephyr/StreamManager.cuh&quot;</span></div>
<div class="line"><a id="l00003" name="l00003"></a><span class="lineno">    3</span><span class="preprocessor">#include &lt;filesystem&gt;</span></div>
<div class="line"><a id="l00004" name="l00004"></a><span class="lineno">    4</span><span class="preprocessor">#include &lt;sstream&gt;</span></div>
<div class="line"><a id="l00005" name="l00005"></a><span class="lineno">    5</span><span class="preprocessor">#include &lt;fstream&gt;</span></div>
<div class="line"><a id="l00006" name="l00006"></a><span class="lineno">    6</span> </div>
<div class="line"><a id="l00007" name="l00007"></a><span class="lineno">    7</span><span class="keyword">namespace </span><a class="code hl_namespace" href="namespacenz_1_1graph.html">nz::graph</a> {</div>
<div class="foldopen" id="foldopen00008" data-start="{" data-end="}">
<div class="line"><a id="l00008" name="l00008"></a><span class="lineno"><a class="line" href="classnz_1_1graph_1_1_compute_graph.html#a1d54fe96c53fec967c8a20e1ac79ceed">    8</a></span>    std::ostream&amp; <a class="code hl_function" href="classnz_1_1graph_1_1_compute_graph.html#a1d54fe96c53fec967c8a20e1ac79ceed">ComputeGraph::print</a>(std::ostream&amp; os) {</div>
<div class="line"><a id="l00009" name="l00009"></a><span class="lineno">    9</span>        <span class="keywordflow">if</span> (!<a class="code hl_function" href="classnz_1_1graph_1_1_compute_graph.html#aede9d05b6c8e7394ea730b5ffea42164">isSorted</a>()) {</div>
<div class="line"><a id="l00010" name="l00010"></a><span class="lineno">   10</span>            <a class="code hl_function" href="classnz_1_1graph_1_1_compute_graph.html#a5960e75d631200994c0a2f78f58674dd">topologicalSort</a>();</div>
<div class="line"><a id="l00011" name="l00011"></a><span class="lineno">   11</span>        }</div>
<div class="line"><a id="l00012" name="l00012"></a><span class="lineno">   12</span>        <span class="keywordflow">for</span> (<a class="code hl_class" href="classnz_1_1nodes_1_1_node.html">Node</a>* node : sortedNodes) {</div>
<div class="line"><a id="l00013" name="l00013"></a><span class="lineno">   13</span>            os &lt;&lt; <span class="stringliteral">&quot;Node:&quot;</span> &lt;&lt; nodeRosterReverse[node] &lt;&lt; <span class="stringliteral">&quot;\n&quot;</span>;</div>
<div class="line"><a id="l00014" name="l00014"></a><span class="lineno">   14</span>            os &lt;&lt; <span class="stringliteral">&quot;Pre:&quot;</span>;</div>
<div class="line"><a id="l00015" name="l00015"></a><span class="lineno">   15</span>            <span class="keywordflow">for</span> (<a class="code hl_class" href="classnz_1_1nodes_1_1_node.html">Node</a>* pre : node-&gt;inputs) {</div>
<div class="line"><a id="l00016" name="l00016"></a><span class="lineno">   16</span>                os &lt;&lt; <span class="stringliteral">&quot; &quot;</span> &lt;&lt; nodeRosterReverse[pre];</div>
<div class="line"><a id="l00017" name="l00017"></a><span class="lineno">   17</span>            }</div>
<div class="line"><a id="l00018" name="l00018"></a><span class="lineno">   18</span>            os &lt;&lt; <span class="stringliteral">&quot; || &quot;</span>;</div>
<div class="line"><a id="l00019" name="l00019"></a><span class="lineno">   19</span>            os &lt;&lt; <span class="stringliteral">&quot;Next:&quot;</span>;</div>
<div class="line"><a id="l00020" name="l00020"></a><span class="lineno">   20</span>            <span class="keywordflow">for</span> (<a class="code hl_class" href="classnz_1_1nodes_1_1_node.html">Node</a>* next : adjList[node]) {</div>
<div class="line"><a id="l00021" name="l00021"></a><span class="lineno">   21</span>                os &lt;&lt; <span class="stringliteral">&quot; &quot;</span> &lt;&lt; nodeRosterReverse[next];</div>
<div class="line"><a id="l00022" name="l00022"></a><span class="lineno">   22</span>            }</div>
<div class="line"><a id="l00023" name="l00023"></a><span class="lineno">   23</span>            os &lt;&lt; <span class="stringliteral">&quot;\n&quot;</span>;</div>
<div class="line"><a id="l00024" name="l00024"></a><span class="lineno">   24</span>            os &lt;&lt; <span class="stringliteral">&quot;Data:\n&quot;</span>;</div>
<div class="line"><a id="l00025" name="l00025"></a><span class="lineno">   25</span>            os &lt;&lt; *node-&gt;output;</div>
<div class="line"><a id="l00026" name="l00026"></a><span class="lineno">   26</span>            os &lt;&lt; <span class="stringliteral">&quot;\n&quot;</span>;</div>
<div class="line"><a id="l00027" name="l00027"></a><span class="lineno">   27</span>            <span class="keywordflow">if</span> (node-&gt;output-&gt;requiresGrad()) {</div>
<div class="line"><a id="l00028" name="l00028"></a><span class="lineno">   28</span>                os &lt;&lt; <span class="stringliteral">&quot;Grad:\n&quot;</span>;</div>
<div class="line"><a id="l00029" name="l00029"></a><span class="lineno">   29</span>                node-&gt;output-&gt;printGrad(os);</div>
<div class="line"><a id="l00030" name="l00030"></a><span class="lineno">   30</span>                os &lt;&lt; <span class="stringliteral">&quot;\n&quot;</span>;</div>
<div class="line"><a id="l00031" name="l00031"></a><span class="lineno">   31</span>            }</div>
<div class="line"><a id="l00032" name="l00032"></a><span class="lineno">   32</span>        }</div>
<div class="line"><a id="l00033" name="l00033"></a><span class="lineno">   33</span>        os &lt;&lt; <span class="stringliteral">&quot;loss: &quot;</span> &lt;&lt; outputNodes[0]-&gt;getLoss() &lt;&lt; <span class="stringliteral">&quot;\n&quot;</span>;</div>
<div class="line"><a id="l00034" name="l00034"></a><span class="lineno">   34</span>        <span class="keywordflow">return</span> os;</div>
<div class="line"><a id="l00035" name="l00035"></a><span class="lineno">   35</span>    }</div>
</div>
<div class="line"><a id="l00036" name="l00036"></a><span class="lineno">   36</span> </div>
<div class="foldopen" id="foldopen00056" data-start="{" data-end="};">
<div class="line"><a id="l00056" name="l00056"></a><span class="lineno"><a class="line" href="classnz_1_1graph_1_1_compute_graph.html#a25cf602885275406f8fe60a5077308a8">   56</a></span>    std::ostream&amp; <a class="code hl_function" href="namespacenz_1_1nodes.html#a42672c2d7708ae1d0c071fd9bef6c03c">operator&lt;&lt;</a>(std::ostream&amp; os, <a class="code hl_class" href="classnz_1_1graph_1_1_compute_graph.html">ComputeGraph</a>&amp; graph) {</div>
<div class="line"><a id="l00057" name="l00057"></a><span class="lineno">   57</span>        <span class="keywordflow">return</span> graph.<a class="code hl_function" href="classnz_1_1graph_1_1_compute_graph.html#a1d54fe96c53fec967c8a20e1ac79ceed">print</a>(os);</div>
<div class="line"><a id="l00058" name="l00058"></a><span class="lineno">   58</span>    }</div>
</div>
<div class="line"><a id="l00059" name="l00059"></a><span class="lineno">   59</span> </div>
<div class="foldopen" id="foldopen00108" data-start="{" data-end="};">
<div class="line"><a id="l00108" name="l00108"></a><span class="lineno"><a class="line" href="classnz_1_1graph_1_1_compute_graph.html#a11f567efba2c857e64a88c411c5e6c54">  108</a></span>    <span class="keywordtype">void</span> <a class="code hl_function" href="namespacenz_1_1graph.html#a3c6187d695022b05df227d913ac89d91">CreateNode</a>(<a class="code hl_class" href="classnz_1_1graph_1_1_compute_graph.html">ComputeGraph</a>* graph, <span class="keyword">const</span> std::string&amp; type, <span class="keyword">const</span> std::string&amp; name, std::vector&lt;int&gt; pre,</div>
<div class="line"><a id="l00109" name="l00109"></a><span class="lineno">  109</span>                    <span class="keyword">const</span> <a class="code hl_class" href="classnz_1_1data_1_1_dimension.html">Tensor::shape_type</a>&amp; shape, <span class="keywordtype">float</span>* data, <span class="keyword">const</span> <span class="keywordtype">bool</span> requires_grad, <span class="keywordtype">float</span>* grad) {</div>
<div class="line"><a id="l00110" name="l00110"></a><span class="lineno">  110</span>        <span class="keywordflow">if</span> (type == <span class="stringliteral">&quot;Input&quot;</span>) {</div>
<div class="line"><a id="l00111" name="l00111"></a><span class="lineno">  111</span>            <span class="keyword">auto</span>* inputNode = <span class="keyword">new</span> <a class="code hl_class" href="classnz_1_1nodes_1_1io_1_1_input_node.html">InputNode</a>(shape, requires_grad);</div>
<div class="line"><a id="l00112" name="l00112"></a><span class="lineno">  112</span>            inputNode-&gt;output-&gt;dataInject(data);</div>
<div class="line"><a id="l00113" name="l00113"></a><span class="lineno">  113</span>            <span class="keywordflow">if</span> (requires_grad) {</div>
<div class="line"><a id="l00114" name="l00114"></a><span class="lineno">  114</span>                inputNode-&gt;output-&gt;dataInject(grad, <span class="keyword">true</span>);</div>
<div class="line"><a id="l00115" name="l00115"></a><span class="lineno">  115</span>            }</div>
<div class="line"><a id="l00116" name="l00116"></a><span class="lineno">  116</span>            graph-&gt;<a class="code hl_function" href="classnz_1_1graph_1_1_compute_graph.html#a5ce9a271dcea6cc2b9490cd815ee53ed">addNode</a>(inputNode, name);</div>
<div class="line"><a id="l00117" name="l00117"></a><span class="lineno">  117</span>            graph-&gt;sortedNodes.push_back(inputNode);</div>
<div class="line"><a id="l00118" name="l00118"></a><span class="lineno">  118</span>            graph-&gt;inputNodes.push_back(inputNode);</div>
<div class="line"><a id="l00119" name="l00119"></a><span class="lineno">  119</span>        }</div>
<div class="line"><a id="l00120" name="l00120"></a><span class="lineno">  120</span>        <span class="keywordflow">else</span> <span class="keywordflow">if</span> (type == <span class="stringliteral">&quot;Output&quot;</span>) {</div>
<div class="line"><a id="l00121" name="l00121"></a><span class="lineno">  121</span>            <span class="keyword">auto</span>* outputNode = <span class="keyword">new</span> <a class="code hl_class" href="classnz_1_1nodes_1_1io_1_1_output_node.html">OutputNode</a>(graph-&gt;sortedNodes[pre[0]]);</div>
<div class="line"><a id="l00122" name="l00122"></a><span class="lineno">  122</span>            outputNode-&gt;forward();</div>
<div class="line"><a id="l00123" name="l00123"></a><span class="lineno">  123</span>            graph-&gt;<a class="code hl_function" href="classnz_1_1graph_1_1_compute_graph.html#a5ce9a271dcea6cc2b9490cd815ee53ed">addNode</a>(outputNode, name);</div>
<div class="line"><a id="l00124" name="l00124"></a><span class="lineno">  124</span>            graph-&gt;outputNodes.push_back(outputNode);</div>
<div class="line"><a id="l00125" name="l00125"></a><span class="lineno">  125</span>            graph-&gt;sortedNodes.push_back(outputNode);</div>
<div class="line"><a id="l00126" name="l00126"></a><span class="lineno">  126</span>        }</div>
<div class="line"><a id="l00127" name="l00127"></a><span class="lineno">  127</span>        <span class="keywordflow">else</span> <span class="keywordflow">if</span> (type == <span class="stringliteral">&quot;Add&quot;</span>) {</div>
<div class="line"><a id="l00128" name="l00128"></a><span class="lineno">  128</span>            <span class="keyword">auto</span>* addNode = <span class="keyword">new</span> <a class="code hl_class" href="classnz_1_1nodes_1_1calc_1_1_add_node.html">AddNode</a>(graph-&gt;sortedNodes[pre[0]], graph-&gt;sortedNodes[pre[1]]);</div>
<div class="line"><a id="l00129" name="l00129"></a><span class="lineno">  129</span>            addNode-&gt;output-&gt;dataInject(data);</div>
<div class="line"><a id="l00130" name="l00130"></a><span class="lineno">  130</span>            <span class="keywordflow">if</span> (requires_grad) {</div>
<div class="line"><a id="l00131" name="l00131"></a><span class="lineno">  131</span>                addNode-&gt;output-&gt;dataInject(grad, <span class="keyword">true</span>);</div>
<div class="line"><a id="l00132" name="l00132"></a><span class="lineno">  132</span>            }</div>
<div class="line"><a id="l00133" name="l00133"></a><span class="lineno">  133</span>            graph-&gt;<a class="code hl_function" href="classnz_1_1graph_1_1_compute_graph.html#a5ce9a271dcea6cc2b9490cd815ee53ed">addNode</a>(addNode, name);</div>
<div class="line"><a id="l00134" name="l00134"></a><span class="lineno">  134</span>            graph-&gt;sortedNodes.push_back(addNode);</div>
<div class="line"><a id="l00135" name="l00135"></a><span class="lineno">  135</span>        }</div>
<div class="line"><a id="l00136" name="l00136"></a><span class="lineno">  136</span>        <span class="keywordflow">else</span> <span class="keywordflow">if</span> (type == <span class="stringliteral">&quot;MatMul&quot;</span>) {</div>
<div class="line"><a id="l00137" name="l00137"></a><span class="lineno">  137</span>            <span class="keyword">auto</span>* matmulNode = <span class="keyword">new</span> <a class="code hl_class" href="classnz_1_1nodes_1_1calc_1_1_mat_mul_node.html">MatMulNode</a>(graph-&gt;sortedNodes[pre[0]], graph-&gt;sortedNodes[pre[1]]);</div>
<div class="line"><a id="l00138" name="l00138"></a><span class="lineno">  138</span>            matmulNode-&gt;output-&gt;dataInject(data);</div>
<div class="line"><a id="l00139" name="l00139"></a><span class="lineno">  139</span>            <span class="keywordflow">if</span> (requires_grad) {</div>
<div class="line"><a id="l00140" name="l00140"></a><span class="lineno">  140</span>                matmulNode-&gt;output-&gt;dataInject(grad, <span class="keyword">true</span>);</div>
<div class="line"><a id="l00141" name="l00141"></a><span class="lineno">  141</span>            }</div>
<div class="line"><a id="l00142" name="l00142"></a><span class="lineno">  142</span>            graph-&gt;<a class="code hl_function" href="classnz_1_1graph_1_1_compute_graph.html#a5ce9a271dcea6cc2b9490cd815ee53ed">addNode</a>(matmulNode, name);</div>
<div class="line"><a id="l00143" name="l00143"></a><span class="lineno">  143</span>            graph-&gt;sortedNodes.push_back(matmulNode);</div>
<div class="line"><a id="l00144" name="l00144"></a><span class="lineno">  144</span>        }</div>
<div class="line"><a id="l00145" name="l00145"></a><span class="lineno">  145</span>        <span class="keywordflow">else</span> <span class="keywordflow">if</span> (type == <span class="stringliteral">&quot;ScalarMul&quot;</span> || type == <span class="stringliteral">&quot;ScalarDiv&quot;</span> || type == <span class="stringliteral">&quot;ScalarAdd&quot;</span> || type == <span class="stringliteral">&quot;ScalarSub&quot;</span>) {</div>
<div class="line"><a id="l00146" name="l00146"></a><span class="lineno">  146</span>            <span class="keywordflow">throw</span> std::runtime_error(<span class="stringliteral">&quot;Scalar operations not supported&quot;</span>);</div>
<div class="line"><a id="l00147" name="l00147"></a><span class="lineno">  147</span>        }</div>
<div class="line"><a id="l00148" name="l00148"></a><span class="lineno">  148</span>        <span class="keywordflow">else</span> <span class="keywordflow">if</span> (type == <span class="stringliteral">&quot;Sub&quot;</span>) {</div>
<div class="line"><a id="l00149" name="l00149"></a><span class="lineno">  149</span>            <span class="keyword">auto</span>* subNode = <span class="keyword">new</span> <a class="code hl_class" href="classnz_1_1nodes_1_1calc_1_1_sub_node.html">SubNode</a>(graph-&gt;sortedNodes[pre[0]], graph-&gt;sortedNodes[pre[1]]);</div>
<div class="line"><a id="l00150" name="l00150"></a><span class="lineno">  150</span>            subNode-&gt;output-&gt;dataInject(data);</div>
<div class="line"><a id="l00151" name="l00151"></a><span class="lineno">  151</span>            <span class="keywordflow">if</span> (requires_grad) {</div>
<div class="line"><a id="l00152" name="l00152"></a><span class="lineno">  152</span>                subNode-&gt;output-&gt;dataInject(grad, <span class="keyword">true</span>);</div>
<div class="line"><a id="l00153" name="l00153"></a><span class="lineno">  153</span>            }</div>
<div class="line"><a id="l00154" name="l00154"></a><span class="lineno">  154</span>            graph-&gt;<a class="code hl_function" href="classnz_1_1graph_1_1_compute_graph.html#a5ce9a271dcea6cc2b9490cd815ee53ed">addNode</a>(subNode, name);</div>
<div class="line"><a id="l00155" name="l00155"></a><span class="lineno">  155</span>            graph-&gt;sortedNodes.push_back(subNode);</div>
<div class="line"><a id="l00156" name="l00156"></a><span class="lineno">  156</span>        }</div>
<div class="line"><a id="l00157" name="l00157"></a><span class="lineno">  157</span>        <span class="keywordflow">else</span> <span class="keywordflow">if</span> (type == <span class="stringliteral">&quot;ReLU&quot;</span>) {</div>
<div class="line"><a id="l00158" name="l00158"></a><span class="lineno">  158</span>            <span class="keyword">auto</span>* reluNode = <span class="keyword">new</span> <a class="code hl_class" href="classnz_1_1nodes_1_1calc_1_1_re_l_u_node.html">ReLUNode</a>(graph-&gt;sortedNodes[pre[0]]);</div>
<div class="line"><a id="l00159" name="l00159"></a><span class="lineno">  159</span>            reluNode-&gt;output-&gt;dataInject(data);</div>
<div class="line"><a id="l00160" name="l00160"></a><span class="lineno">  160</span>            <span class="keywordflow">if</span> (requires_grad) {</div>
<div class="line"><a id="l00161" name="l00161"></a><span class="lineno">  161</span>                reluNode-&gt;output-&gt;dataInject(grad, <span class="keyword">true</span>);</div>
<div class="line"><a id="l00162" name="l00162"></a><span class="lineno">  162</span>            }</div>
<div class="line"><a id="l00163" name="l00163"></a><span class="lineno">  163</span>            graph-&gt;<a class="code hl_function" href="classnz_1_1graph_1_1_compute_graph.html#a5ce9a271dcea6cc2b9490cd815ee53ed">addNode</a>(reluNode, name);</div>
<div class="line"><a id="l00164" name="l00164"></a><span class="lineno">  164</span>            graph-&gt;sortedNodes.push_back(reluNode);</div>
<div class="line"><a id="l00165" name="l00165"></a><span class="lineno">  165</span>        }</div>
<div class="line"><a id="l00166" name="l00166"></a><span class="lineno">  166</span>        <span class="keywordflow">else</span> <span class="keywordflow">if</span> (type == <span class="stringliteral">&quot;Sigmoid&quot;</span>) {</div>
<div class="line"><a id="l00167" name="l00167"></a><span class="lineno">  167</span>            <span class="keyword">auto</span>* sigmoidNode = <span class="keyword">new</span> <a class="code hl_class" href="classnz_1_1nodes_1_1calc_1_1_sigmoid_node.html">SigmoidNode</a>(graph-&gt;sortedNodes[pre[0]]);</div>
<div class="line"><a id="l00168" name="l00168"></a><span class="lineno">  168</span>            sigmoidNode-&gt;output-&gt;dataInject(data);</div>
<div class="line"><a id="l00169" name="l00169"></a><span class="lineno">  169</span>            <span class="keywordflow">if</span> (requires_grad) {</div>
<div class="line"><a id="l00170" name="l00170"></a><span class="lineno">  170</span>                sigmoidNode-&gt;output-&gt;dataInject(grad, <span class="keyword">true</span>);</div>
<div class="line"><a id="l00171" name="l00171"></a><span class="lineno">  171</span>            }</div>
<div class="line"><a id="l00172" name="l00172"></a><span class="lineno">  172</span>            graph-&gt;<a class="code hl_function" href="classnz_1_1graph_1_1_compute_graph.html#a5ce9a271dcea6cc2b9490cd815ee53ed">addNode</a>(sigmoidNode, name);</div>
<div class="line"><a id="l00173" name="l00173"></a><span class="lineno">  173</span>            graph-&gt;sortedNodes.push_back(sigmoidNode);</div>
<div class="line"><a id="l00174" name="l00174"></a><span class="lineno">  174</span>        }</div>
<div class="line"><a id="l00175" name="l00175"></a><span class="lineno">  175</span>        <span class="keywordflow">else</span> <span class="keywordflow">if</span> (type == <span class="stringliteral">&quot;Tanh&quot;</span>) {</div>
<div class="line"><a id="l00176" name="l00176"></a><span class="lineno">  176</span>            <span class="keyword">auto</span>* tanhNode = <span class="keyword">new</span> <a class="code hl_class" href="classnz_1_1nodes_1_1calc_1_1_tanh_node.html">TanhNode</a>(graph-&gt;sortedNodes[pre[0]]);</div>
<div class="line"><a id="l00177" name="l00177"></a><span class="lineno">  177</span>            tanhNode-&gt;output-&gt;dataInject(data);</div>
<div class="line"><a id="l00178" name="l00178"></a><span class="lineno">  178</span>            <span class="keywordflow">if</span> (requires_grad) {</div>
<div class="line"><a id="l00179" name="l00179"></a><span class="lineno">  179</span>                tanhNode-&gt;output-&gt;dataInject(grad, <span class="keyword">true</span>);</div>
<div class="line"><a id="l00180" name="l00180"></a><span class="lineno">  180</span>            }</div>
<div class="line"><a id="l00181" name="l00181"></a><span class="lineno">  181</span>            graph-&gt;<a class="code hl_function" href="classnz_1_1graph_1_1_compute_graph.html#a5ce9a271dcea6cc2b9490cd815ee53ed">addNode</a>(tanhNode, name);</div>
<div class="line"><a id="l00182" name="l00182"></a><span class="lineno">  182</span>            graph-&gt;sortedNodes.push_back(tanhNode);</div>
<div class="line"><a id="l00183" name="l00183"></a><span class="lineno">  183</span>        }</div>
<div class="line"><a id="l00184" name="l00184"></a><span class="lineno">  184</span>        <span class="keywordflow">else</span> <span class="keywordflow">if</span> (type == <span class="stringliteral">&quot;LeakyReLU&quot;</span>) {</div>
<div class="line"><a id="l00185" name="l00185"></a><span class="lineno">  185</span>            <span class="keyword">auto</span>* leakyreluNode = <span class="keyword">new</span> <a class="code hl_class" href="classnz_1_1nodes_1_1calc_1_1_leaky_re_l_u_node.html">LeakyReLUNode</a>(graph-&gt;sortedNodes[pre[0]]);</div>
<div class="line"><a id="l00186" name="l00186"></a><span class="lineno">  186</span>            leakyreluNode-&gt;output-&gt;dataInject(data);</div>
<div class="line"><a id="l00187" name="l00187"></a><span class="lineno">  187</span>            <span class="keywordflow">if</span> (requires_grad) {</div>
<div class="line"><a id="l00188" name="l00188"></a><span class="lineno">  188</span>                leakyreluNode-&gt;output-&gt;dataInject(grad, <span class="keyword">true</span>);</div>
<div class="line"><a id="l00189" name="l00189"></a><span class="lineno">  189</span>            }</div>
<div class="line"><a id="l00190" name="l00190"></a><span class="lineno">  190</span>            graph-&gt;<a class="code hl_function" href="classnz_1_1graph_1_1_compute_graph.html#a5ce9a271dcea6cc2b9490cd815ee53ed">addNode</a>(leakyreluNode, name);</div>
<div class="line"><a id="l00191" name="l00191"></a><span class="lineno">  191</span>            graph-&gt;sortedNodes.push_back(leakyreluNode);</div>
<div class="line"><a id="l00192" name="l00192"></a><span class="lineno">  192</span>        }</div>
<div class="line"><a id="l00193" name="l00193"></a><span class="lineno">  193</span>        <span class="keywordflow">else</span> <span class="keywordflow">if</span> (type == <span class="stringliteral">&quot;Swish&quot;</span>) {</div>
<div class="line"><a id="l00194" name="l00194"></a><span class="lineno">  194</span>            <span class="keyword">auto</span>* swishNode = <span class="keyword">new</span> <a class="code hl_class" href="classnz_1_1nodes_1_1calc_1_1_swish_node.html">SwishNode</a>(graph-&gt;sortedNodes[pre[0]]);</div>
<div class="line"><a id="l00195" name="l00195"></a><span class="lineno">  195</span>            swishNode-&gt;output-&gt;dataInject(data);</div>
<div class="line"><a id="l00196" name="l00196"></a><span class="lineno">  196</span>            <span class="keywordflow">if</span> (requires_grad) {</div>
<div class="line"><a id="l00197" name="l00197"></a><span class="lineno">  197</span>                swishNode-&gt;output-&gt;dataInject(grad, <span class="keyword">true</span>);</div>
<div class="line"><a id="l00198" name="l00198"></a><span class="lineno">  198</span>            }</div>
<div class="line"><a id="l00199" name="l00199"></a><span class="lineno">  199</span>            graph-&gt;<a class="code hl_function" href="classnz_1_1graph_1_1_compute_graph.html#a5ce9a271dcea6cc2b9490cd815ee53ed">addNode</a>(swishNode, name);</div>
<div class="line"><a id="l00200" name="l00200"></a><span class="lineno">  200</span>            graph-&gt;sortedNodes.push_back(swishNode);</div>
<div class="line"><a id="l00201" name="l00201"></a><span class="lineno">  201</span>        }</div>
<div class="line"><a id="l00202" name="l00202"></a><span class="lineno">  202</span>        <span class="keywordflow">else</span> <span class="keywordflow">if</span> (type == <span class="stringliteral">&quot;ELU&quot;</span>) {</div>
<div class="line"><a id="l00203" name="l00203"></a><span class="lineno">  203</span>            <span class="keyword">auto</span>* eluNode = <span class="keyword">new</span> <a class="code hl_class" href="classnz_1_1nodes_1_1calc_1_1_e_l_u_node.html">ELUNode</a>(graph-&gt;sortedNodes[pre[0]]);</div>
<div class="line"><a id="l00204" name="l00204"></a><span class="lineno">  204</span>            eluNode-&gt;output-&gt;dataInject(data);</div>
<div class="line"><a id="l00205" name="l00205"></a><span class="lineno">  205</span>            <span class="keywordflow">if</span> (requires_grad) {</div>
<div class="line"><a id="l00206" name="l00206"></a><span class="lineno">  206</span>                eluNode-&gt;output-&gt;dataInject(grad, <span class="keyword">true</span>);</div>
<div class="line"><a id="l00207" name="l00207"></a><span class="lineno">  207</span>            }</div>
<div class="line"><a id="l00208" name="l00208"></a><span class="lineno">  208</span>            graph-&gt;<a class="code hl_function" href="classnz_1_1graph_1_1_compute_graph.html#a5ce9a271dcea6cc2b9490cd815ee53ed">addNode</a>(eluNode);</div>
<div class="line"><a id="l00209" name="l00209"></a><span class="lineno">  209</span>            graph-&gt;sortedNodes.push_back(eluNode);</div>
<div class="line"><a id="l00210" name="l00210"></a><span class="lineno">  210</span>        }</div>
<div class="line"><a id="l00211" name="l00211"></a><span class="lineno">  211</span>        <span class="keywordflow">else</span> <span class="keywordflow">if</span> (type == <span class="stringliteral">&quot;HardSigmoid&quot;</span>) {</div>
<div class="line"><a id="l00212" name="l00212"></a><span class="lineno">  212</span>            <span class="keyword">auto</span>* hardsigmoidNode = <span class="keyword">new</span> <a class="code hl_class" href="classnz_1_1nodes_1_1calc_1_1_hard_sigmoid_node.html">HardSigmoidNode</a>(graph-&gt;sortedNodes[pre[0]]);</div>
<div class="line"><a id="l00213" name="l00213"></a><span class="lineno">  213</span>            hardsigmoidNode-&gt;output-&gt;dataInject(data);</div>
<div class="line"><a id="l00214" name="l00214"></a><span class="lineno">  214</span>            <span class="keywordflow">if</span> (requires_grad) {</div>
<div class="line"><a id="l00215" name="l00215"></a><span class="lineno">  215</span>                hardsigmoidNode-&gt;output-&gt;dataInject(grad, <span class="keyword">true</span>);</div>
<div class="line"><a id="l00216" name="l00216"></a><span class="lineno">  216</span>            }</div>
<div class="line"><a id="l00217" name="l00217"></a><span class="lineno">  217</span>            graph-&gt;<a class="code hl_function" href="classnz_1_1graph_1_1_compute_graph.html#a5ce9a271dcea6cc2b9490cd815ee53ed">addNode</a>(hardsigmoidNode, name);</div>
<div class="line"><a id="l00218" name="l00218"></a><span class="lineno">  218</span>            graph-&gt;sortedNodes.push_back(hardsigmoidNode);</div>
<div class="line"><a id="l00219" name="l00219"></a><span class="lineno">  219</span>        }</div>
<div class="line"><a id="l00220" name="l00220"></a><span class="lineno">  220</span>        <span class="keywordflow">else</span> <span class="keywordflow">if</span> (type == <span class="stringliteral">&quot;HardSwish&quot;</span>) {</div>
<div class="line"><a id="l00221" name="l00221"></a><span class="lineno">  221</span>            <span class="keyword">auto</span>* hardswishNode = <span class="keyword">new</span> <a class="code hl_class" href="classnz_1_1nodes_1_1calc_1_1_hard_swish_node.html">HardSwishNode</a>(graph-&gt;sortedNodes[pre[0]]);</div>
<div class="line"><a id="l00222" name="l00222"></a><span class="lineno">  222</span>            hardswishNode-&gt;output-&gt;dataInject(data);</div>
<div class="line"><a id="l00223" name="l00223"></a><span class="lineno">  223</span>            <span class="keywordflow">if</span> (requires_grad) {</div>
<div class="line"><a id="l00224" name="l00224"></a><span class="lineno">  224</span>                hardswishNode-&gt;output-&gt;dataInject(grad, <span class="keyword">true</span>);</div>
<div class="line"><a id="l00225" name="l00225"></a><span class="lineno">  225</span>            }</div>
<div class="line"><a id="l00226" name="l00226"></a><span class="lineno">  226</span>            graph-&gt;<a class="code hl_function" href="classnz_1_1graph_1_1_compute_graph.html#a5ce9a271dcea6cc2b9490cd815ee53ed">addNode</a>(hardswishNode, name);</div>
<div class="line"><a id="l00227" name="l00227"></a><span class="lineno">  227</span>            graph-&gt;sortedNodes.push_back(hardswishNode);</div>
<div class="line"><a id="l00228" name="l00228"></a><span class="lineno">  228</span>        }</div>
<div class="line"><a id="l00229" name="l00229"></a><span class="lineno">  229</span>        <span class="keywordflow">else</span> <span class="keywordflow">if</span> (type == <span class="stringliteral">&quot;Softmax&quot;</span>) {</div>
<div class="line"><a id="l00230" name="l00230"></a><span class="lineno">  230</span>            <span class="keyword">auto</span>* softmaxNode = <span class="keyword">new</span> <a class="code hl_class" href="classnz_1_1nodes_1_1calc_1_1_softmax_node.html">SoftmaxNode</a>(graph-&gt;sortedNodes[pre[0]]);</div>
<div class="line"><a id="l00231" name="l00231"></a><span class="lineno">  231</span>            softmaxNode-&gt;output-&gt;dataInject(data);</div>
<div class="line"><a id="l00232" name="l00232"></a><span class="lineno">  232</span>            <span class="keywordflow">if</span> (requires_grad) {</div>
<div class="line"><a id="l00233" name="l00233"></a><span class="lineno">  233</span>                softmaxNode-&gt;output-&gt;dataInject(grad, <span class="keyword">true</span>);</div>
<div class="line"><a id="l00234" name="l00234"></a><span class="lineno">  234</span>            }</div>
<div class="line"><a id="l00235" name="l00235"></a><span class="lineno">  235</span>            graph-&gt;<a class="code hl_function" href="classnz_1_1graph_1_1_compute_graph.html#a5ce9a271dcea6cc2b9490cd815ee53ed">addNode</a>(softmaxNode, name);</div>
<div class="line"><a id="l00236" name="l00236"></a><span class="lineno">  236</span>            graph-&gt;sortedNodes.push_back(softmaxNode);</div>
<div class="line"><a id="l00237" name="l00237"></a><span class="lineno">  237</span>        }</div>
<div class="line"><a id="l00238" name="l00238"></a><span class="lineno">  238</span>        <span class="keywordflow">else</span> <span class="keywordflow">if</span> (type == <span class="stringliteral">&quot;MeanSquaredError&quot;</span>) {</div>
<div class="line"><a id="l00239" name="l00239"></a><span class="lineno">  239</span>            <span class="keyword">auto</span>* mseNode = <span class="keyword">new</span> <a class="code hl_class" href="classnz_1_1nodes_1_1loss_1_1_mean_squared_error_node.html">MeanSquaredErrorNode</a>(graph-&gt;sortedNodes[pre[0]], graph-&gt;sortedNodes[pre[1]]);</div>
<div class="line"><a id="l00240" name="l00240"></a><span class="lineno">  240</span>            mseNode-&gt;forward();</div>
<div class="line"><a id="l00241" name="l00241"></a><span class="lineno">  241</span>            graph-&gt;<a class="code hl_function" href="classnz_1_1graph_1_1_compute_graph.html#a5ce9a271dcea6cc2b9490cd815ee53ed">addNode</a>(mseNode, name);</div>
<div class="line"><a id="l00242" name="l00242"></a><span class="lineno">  242</span>            graph-&gt;sortedNodes.push_back(mseNode);</div>
<div class="line"><a id="l00243" name="l00243"></a><span class="lineno">  243</span>            graph-&gt;outputNodes.push_back(mseNode);</div>
<div class="line"><a id="l00244" name="l00244"></a><span class="lineno">  244</span>        }</div>
<div class="line"><a id="l00245" name="l00245"></a><span class="lineno">  245</span>        <span class="keywordflow">else</span> <span class="keywordflow">if</span> (type == <span class="stringliteral">&quot;BinaryCrossEntropy&quot;</span>) {</div>
<div class="line"><a id="l00246" name="l00246"></a><span class="lineno">  246</span>            <span class="keyword">auto</span>* bceNode = <span class="keyword">new</span> <a class="code hl_class" href="classnz_1_1nodes_1_1loss_1_1_binary_cross_entropy_node.html">BinaryCrossEntropyNode</a>(graph-&gt;sortedNodes[pre[0]], graph-&gt;sortedNodes[pre[1]]);</div>
<div class="line"><a id="l00247" name="l00247"></a><span class="lineno">  247</span>            bceNode-&gt;forward();</div>
<div class="line"><a id="l00248" name="l00248"></a><span class="lineno">  248</span>            graph-&gt;<a class="code hl_function" href="classnz_1_1graph_1_1_compute_graph.html#a5ce9a271dcea6cc2b9490cd815ee53ed">addNode</a>(bceNode, name);</div>
<div class="line"><a id="l00249" name="l00249"></a><span class="lineno">  249</span>            graph-&gt;sortedNodes.push_back(bceNode);</div>
<div class="line"><a id="l00250" name="l00250"></a><span class="lineno">  250</span>            graph-&gt;outputNodes.push_back(bceNode);</div>
<div class="line"><a id="l00251" name="l00251"></a><span class="lineno">  251</span>        }</div>
<div class="line"><a id="l00252" name="l00252"></a><span class="lineno">  252</span>        <span class="keywordflow">else</span> {</div>
<div class="line"><a id="l00253" name="l00253"></a><span class="lineno">  253</span>            <span class="keywordflow">throw</span> std::runtime_error(<span class="stringliteral">&quot;Unknown node type&quot;</span>);</div>
<div class="line"><a id="l00254" name="l00254"></a><span class="lineno">  254</span>        }</div>
<div class="line"><a id="l00255" name="l00255"></a><span class="lineno">  255</span> </div>
<div class="line"><a id="l00256" name="l00256"></a><span class="lineno">  256</span>    }</div>
</div>
<div class="line"><a id="l00257" name="l00257"></a><span class="lineno">  257</span> </div>
<div class="foldopen" id="foldopen00258" data-start="{" data-end="}">
<div class="line"><a id="l00258" name="l00258"></a><span class="lineno"><a class="line" href="classnz_1_1graph_1_1_compute_graph.html#a5960e75d631200994c0a2f78f58674dd">  258</a></span>    <span class="keywordtype">void</span> <a class="code hl_function" href="classnz_1_1graph_1_1_compute_graph.html#a5960e75d631200994c0a2f78f58674dd">ComputeGraph::topologicalSort</a>() {</div>
<div class="line"><a id="l00259" name="l00259"></a><span class="lineno">  259</span>        sortedNodes.clear();</div>
<div class="line"><a id="l00260" name="l00260"></a><span class="lineno">  260</span>        inDegree.clear();</div>
<div class="line"><a id="l00261" name="l00261"></a><span class="lineno">  261</span>        adjList.clear();</div>
<div class="line"><a id="l00262" name="l00262"></a><span class="lineno">  262</span> </div>
<div class="line"><a id="l00263" name="l00263"></a><span class="lineno">  263</span>        <span class="keywordflow">for</span> (<a class="code hl_class" href="classnz_1_1nodes_1_1_node.html">Node</a>* node : nodes) {</div>
<div class="line"><a id="l00264" name="l00264"></a><span class="lineno">  264</span>            <span class="keywordflow">if</span> (inDegree.find(node) == inDegree.end()) {</div>
<div class="line"><a id="l00265" name="l00265"></a><span class="lineno">  265</span>                inDegree[node] = 0;</div>
<div class="line"><a id="l00266" name="l00266"></a><span class="lineno">  266</span>            }</div>
<div class="line"><a id="l00267" name="l00267"></a><span class="lineno">  267</span> </div>
<div class="line"><a id="l00268" name="l00268"></a><span class="lineno">  268</span>            <span class="keywordflow">for</span> (<a class="code hl_class" href="classnz_1_1nodes_1_1_node.html">Node</a>* input : node-&gt;inputs) {</div>
<div class="line"><a id="l00269" name="l00269"></a><span class="lineno">  269</span>                adjList[input].push_back(node);</div>
<div class="line"><a id="l00270" name="l00270"></a><span class="lineno">  270</span>                inDegree[node]++;</div>
<div class="line"><a id="l00271" name="l00271"></a><span class="lineno">  271</span>            }</div>
<div class="line"><a id="l00272" name="l00272"></a><span class="lineno">  272</span>        }</div>
<div class="line"><a id="l00273" name="l00273"></a><span class="lineno">  273</span> </div>
<div class="line"><a id="l00274" name="l00274"></a><span class="lineno">  274</span>        std::queue&lt;Node*&gt; q;</div>
<div class="line"><a id="l00275" name="l00275"></a><span class="lineno">  275</span>        <span class="keywordflow">for</span> (<a class="code hl_class" href="classnz_1_1nodes_1_1_node.html">Node</a>* node : nodes) {</div>
<div class="line"><a id="l00276" name="l00276"></a><span class="lineno">  276</span>            <span class="keywordflow">if</span> (inDegree[node] == 0) {</div>
<div class="line"><a id="l00277" name="l00277"></a><span class="lineno">  277</span>                q.push(node);</div>
<div class="line"><a id="l00278" name="l00278"></a><span class="lineno">  278</span>            }</div>
<div class="line"><a id="l00279" name="l00279"></a><span class="lineno">  279</span>        }</div>
<div class="line"><a id="l00280" name="l00280"></a><span class="lineno">  280</span> </div>
<div class="line"><a id="l00281" name="l00281"></a><span class="lineno">  281</span>        <span class="keywordflow">while</span> (!q.empty()) {</div>
<div class="line"><a id="l00282" name="l00282"></a><span class="lineno">  282</span>            <a class="code hl_class" href="classnz_1_1nodes_1_1_node.html">Node</a>* node = q.front();</div>
<div class="line"><a id="l00283" name="l00283"></a><span class="lineno">  283</span>            q.pop();</div>
<div class="line"><a id="l00284" name="l00284"></a><span class="lineno">  284</span>            sortedNodes.push_back(node);</div>
<div class="line"><a id="l00285" name="l00285"></a><span class="lineno">  285</span>            <span class="keywordflow">for</span> (<a class="code hl_class" href="classnz_1_1nodes_1_1_node.html">Node</a>* next : adjList[node]) {</div>
<div class="line"><a id="l00286" name="l00286"></a><span class="lineno">  286</span>                inDegree[next]--;</div>
<div class="line"><a id="l00287" name="l00287"></a><span class="lineno">  287</span>                <span class="keywordflow">if</span> (inDegree[next] == 0) {</div>
<div class="line"><a id="l00288" name="l00288"></a><span class="lineno">  288</span>                    q.push(next);</div>
<div class="line"><a id="l00289" name="l00289"></a><span class="lineno">  289</span>                }</div>
<div class="line"><a id="l00290" name="l00290"></a><span class="lineno">  290</span>            }</div>
<div class="line"><a id="l00291" name="l00291"></a><span class="lineno">  291</span>        }</div>
<div class="line"><a id="l00292" name="l00292"></a><span class="lineno">  292</span> </div>
<div class="line"><a id="l00293" name="l00293"></a><span class="lineno">  293</span>        <span class="keywordflow">if</span> (sortedNodes.size() != nodes.size()) {</div>
<div class="line"><a id="l00294" name="l00294"></a><span class="lineno">  294</span>            <span class="keywordflow">throw</span> std::runtime_error(<span class="stringliteral">&quot;Graph has cycle&quot;</span>);</div>
<div class="line"><a id="l00295" name="l00295"></a><span class="lineno">  295</span>        }</div>
<div class="line"><a id="l00296" name="l00296"></a><span class="lineno">  296</span>    }</div>
</div>
<div class="line"><a id="l00297" name="l00297"></a><span class="lineno">  297</span> </div>
<div class="foldopen" id="foldopen00298" data-start="{" data-end="}">
<div class="line"><a id="l00298" name="l00298"></a><span class="lineno"><a class="line" href="classnz_1_1graph_1_1_compute_graph.html#aede9d05b6c8e7394ea730b5ffea42164">  298</a></span>    <span class="keywordtype">bool</span> <a class="code hl_function" href="classnz_1_1graph_1_1_compute_graph.html#aede9d05b6c8e7394ea730b5ffea42164">ComputeGraph::isSorted</a>()<span class="keyword"> const </span>{</div>
<div class="line"><a id="l00299" name="l00299"></a><span class="lineno">  299</span>        <span class="keywordflow">return</span> sortedNodes.size() == nodes.size();</div>
<div class="line"><a id="l00300" name="l00300"></a><span class="lineno">  300</span>    }</div>
</div>
<div class="line"><a id="l00301" name="l00301"></a><span class="lineno">  301</span> </div>
<div class="foldopen" id="foldopen00302" data-start="{" data-end="}">
<div class="line"><a id="l00302" name="l00302"></a><span class="lineno"><a class="line" href="classnz_1_1graph_1_1_compute_graph.html#a75a4b72c93448c8617a7af41cc471897">  302</a></span>    <a class="code hl_class" href="classnz_1_1nodes_1_1io_1_1_input_node.html">InputNode</a>* <a class="code hl_function" href="classnz_1_1graph_1_1_compute_graph.html#a75a4b72c93448c8617a7af41cc471897">ComputeGraph::addInput</a>(<span class="keyword">const</span> <a class="code hl_class" href="classnz_1_1data_1_1_dimension.html">Tensor::shape_type</a>&amp; shape, <span class="keywordtype">bool</span> requires_grad, <span class="keyword">const</span> std::string&amp; name) {</div>
<div class="line"><a id="l00303" name="l00303"></a><span class="lineno">  303</span>        <span class="keyword">const</span> <span class="keyword">auto</span> node = <span class="keyword">new</span> <a class="code hl_class" href="classnz_1_1nodes_1_1io_1_1_input_node.html">InputNode</a>(shape, requires_grad);</div>
<div class="line"><a id="l00304" name="l00304"></a><span class="lineno">  304</span>        nodes.push_back(node);</div>
<div class="line"><a id="l00305" name="l00305"></a><span class="lineno">  305</span>        inputNodes.push_back(node);</div>
<div class="line"><a id="l00306" name="l00306"></a><span class="lineno">  306</span>        <span class="keywordflow">if</span> (name == <span class="stringliteral">&quot;default&quot;</span>) {</div>
<div class="line"><a id="l00307" name="l00307"></a><span class="lineno">  307</span>            <span class="keyword">const</span> std::string node_name = node-&gt;type + <span class="stringliteral">&quot;_&quot;</span> + std::to_string(nodesRef);</div>
<div class="line"><a id="l00308" name="l00308"></a><span class="lineno">  308</span>            nodeRoster[node_name] = node;</div>
<div class="line"><a id="l00309" name="l00309"></a><span class="lineno">  309</span>            nodeRosterReverse[node] = node_name;</div>
<div class="line"><a id="l00310" name="l00310"></a><span class="lineno">  310</span>            nodesRef++;</div>
<div class="line"><a id="l00311" name="l00311"></a><span class="lineno">  311</span>        }</div>
<div class="line"><a id="l00312" name="l00312"></a><span class="lineno">  312</span>        <span class="keywordflow">else</span> {</div>
<div class="line"><a id="l00313" name="l00313"></a><span class="lineno">  313</span>            nodeRoster[name] = node;</div>
<div class="line"><a id="l00314" name="l00314"></a><span class="lineno">  314</span>            nodeRosterReverse[node] = name;</div>
<div class="line"><a id="l00315" name="l00315"></a><span class="lineno">  315</span>        }</div>
<div class="line"><a id="l00316" name="l00316"></a><span class="lineno">  316</span>        <span class="keywordflow">return</span> node;</div>
<div class="line"><a id="l00317" name="l00317"></a><span class="lineno">  317</span>    }</div>
</div>
<div class="line"><a id="l00318" name="l00318"></a><span class="lineno">  318</span> </div>
<div class="foldopen" id="foldopen00319" data-start="{" data-end="}">
<div class="line"><a id="l00319" name="l00319"></a><span class="lineno"><a class="line" href="classnz_1_1graph_1_1_compute_graph.html#a86b54a014876f245d64c524dc70075ea">  319</a></span>    <a class="code hl_class" href="classnz_1_1nodes_1_1io_1_1_input_node.html">InputNode</a>* <a class="code hl_function" href="classnz_1_1graph_1_1_compute_graph.html#a75a4b72c93448c8617a7af41cc471897">ComputeGraph::addInput</a>(<span class="keyword">const</span> <a class="code hl_class" href="classnz_1_1data_1_1_tensor.html">Tensor</a>&amp; tensor, <span class="keyword">const</span> std::string&amp; name) {</div>
<div class="line"><a id="l00320" name="l00320"></a><span class="lineno">  320</span>        <span class="keyword">const</span> <span class="keyword">auto</span> node = <span class="keyword">new</span> <a class="code hl_class" href="classnz_1_1nodes_1_1io_1_1_input_node.html">InputNode</a>(tensor);</div>
<div class="line"><a id="l00321" name="l00321"></a><span class="lineno">  321</span>        nodes.push_back(node);</div>
<div class="line"><a id="l00322" name="l00322"></a><span class="lineno">  322</span>        inputNodes.push_back(node);</div>
<div class="line"><a id="l00323" name="l00323"></a><span class="lineno">  323</span>        <span class="keywordflow">if</span> (name == <span class="stringliteral">&quot;default&quot;</span>) {</div>
<div class="line"><a id="l00324" name="l00324"></a><span class="lineno">  324</span>            <span class="keyword">const</span> std::string node_name = node-&gt;type + <span class="stringliteral">&quot;_&quot;</span> + std::to_string(nodesRef);</div>
<div class="line"><a id="l00325" name="l00325"></a><span class="lineno">  325</span>            nodeRoster[node_name] = node;</div>
<div class="line"><a id="l00326" name="l00326"></a><span class="lineno">  326</span>            nodeRosterReverse[node] = node_name;</div>
<div class="line"><a id="l00327" name="l00327"></a><span class="lineno">  327</span>            nodesRef++;</div>
<div class="line"><a id="l00328" name="l00328"></a><span class="lineno">  328</span>        }</div>
<div class="line"><a id="l00329" name="l00329"></a><span class="lineno">  329</span>        <span class="keywordflow">else</span> {</div>
<div class="line"><a id="l00330" name="l00330"></a><span class="lineno">  330</span>            nodeRoster[name] = node;</div>
<div class="line"><a id="l00331" name="l00331"></a><span class="lineno">  331</span>            nodeRosterReverse[node] = name;</div>
<div class="line"><a id="l00332" name="l00332"></a><span class="lineno">  332</span>        }</div>
<div class="line"><a id="l00333" name="l00333"></a><span class="lineno">  333</span>        <span class="keywordflow">return</span> node;</div>
<div class="line"><a id="l00334" name="l00334"></a><span class="lineno">  334</span>    }</div>
</div>
<div class="line"><a id="l00335" name="l00335"></a><span class="lineno">  335</span> </div>
<div class="foldopen" id="foldopen00336" data-start="{" data-end="}">
<div class="line"><a id="l00336" name="l00336"></a><span class="lineno"><a class="line" href="classnz_1_1graph_1_1_compute_graph.html#a8ab70380135ecdeb3652b76c2bf2766c">  336</a></span>    <a class="code hl_class" href="classnz_1_1nodes_1_1io_1_1_input_node.html">InputNode</a>* <a class="code hl_function" href="classnz_1_1graph_1_1_compute_graph.html#a75a4b72c93448c8617a7af41cc471897">ComputeGraph::addInput</a>(<a class="code hl_class" href="classnz_1_1nodes_1_1io_1_1_input_node.html">InputNode</a>* input, <span class="keyword">const</span> std::string&amp; name) {</div>
<div class="line"><a id="l00337" name="l00337"></a><span class="lineno">  337</span>        nodes.push_back(input);</div>
<div class="line"><a id="l00338" name="l00338"></a><span class="lineno">  338</span>        inputNodes.push_back(input);</div>
<div class="line"><a id="l00339" name="l00339"></a><span class="lineno">  339</span>        <span class="keywordflow">if</span> (name == <span class="stringliteral">&quot;default&quot;</span>) {</div>
<div class="line"><a id="l00340" name="l00340"></a><span class="lineno">  340</span>            <span class="keyword">const</span> std::string node_name = input-&gt;type + <span class="stringliteral">&quot;_&quot;</span> + std::to_string(nodesRef);</div>
<div class="line"><a id="l00341" name="l00341"></a><span class="lineno">  341</span>            nodeRoster[node_name] = input;</div>
<div class="line"><a id="l00342" name="l00342"></a><span class="lineno">  342</span>            nodeRosterReverse[input] = node_name;</div>
<div class="line"><a id="l00343" name="l00343"></a><span class="lineno">  343</span>            nodesRef++;</div>
<div class="line"><a id="l00344" name="l00344"></a><span class="lineno">  344</span>        }</div>
<div class="line"><a id="l00345" name="l00345"></a><span class="lineno">  345</span>        <span class="keywordflow">else</span> {</div>
<div class="line"><a id="l00346" name="l00346"></a><span class="lineno">  346</span>            nodeRoster[name] = input;</div>
<div class="line"><a id="l00347" name="l00347"></a><span class="lineno">  347</span>            nodeRosterReverse[input] = name;</div>
<div class="line"><a id="l00348" name="l00348"></a><span class="lineno">  348</span>        }</div>
<div class="line"><a id="l00349" name="l00349"></a><span class="lineno">  349</span>        <span class="keywordflow">return</span> input;</div>
<div class="line"><a id="l00350" name="l00350"></a><span class="lineno">  350</span>    }</div>
</div>
<div class="line"><a id="l00351" name="l00351"></a><span class="lineno">  351</span> </div>
<div class="foldopen" id="foldopen00352" data-start="{" data-end="}">
<div class="line"><a id="l00352" name="l00352"></a><span class="lineno"><a class="line" href="classnz_1_1graph_1_1_compute_graph.html#ab75d30dc8133099118f60bdd07284f01">  352</a></span>    <a class="code hl_class" href="classnz_1_1nodes_1_1io_1_1_input_node.html">InputNode</a>* <a class="code hl_function" href="classnz_1_1graph_1_1_compute_graph.html#a75a4b72c93448c8617a7af41cc471897">ComputeGraph::addInput</a>(<span class="keyword">const</span> <a class="code hl_class" href="classnz_1_1data_1_1_dimension.html">Tensor::shape_type</a>&amp; shape, Tensor::value_type* data,</div>
<div class="line"><a id="l00353" name="l00353"></a><span class="lineno">  353</span>        <span class="keyword">const</span> <span class="keywordtype">bool</span> requires_grad, <span class="keyword">const</span> <span class="keywordtype">bool</span> host, <span class="keyword">const</span> std::string&amp; name) {</div>
<div class="line"><a id="l00354" name="l00354"></a><span class="lineno">  354</span>        <span class="keyword">const</span> <span class="keyword">auto</span> node = <span class="keyword">new</span> <a class="code hl_class" href="classnz_1_1nodes_1_1io_1_1_input_node.html">InputNode</a>(shape, data, requires_grad, host);</div>
<div class="line"><a id="l00355" name="l00355"></a><span class="lineno">  355</span>        nodes.push_back(node);</div>
<div class="line"><a id="l00356" name="l00356"></a><span class="lineno">  356</span>        inputNodes.push_back(node);</div>
<div class="line"><a id="l00357" name="l00357"></a><span class="lineno">  357</span>        <span class="keywordflow">if</span> (name == <span class="stringliteral">&quot;default&quot;</span>) {</div>
<div class="line"><a id="l00358" name="l00358"></a><span class="lineno">  358</span>            <span class="keyword">const</span> std::string node_name = node-&gt;type + <span class="stringliteral">&quot;_&quot;</span> + std::to_string(nodesRef);</div>
<div class="line"><a id="l00359" name="l00359"></a><span class="lineno">  359</span>            nodeRoster[node_name] = node;</div>
<div class="line"><a id="l00360" name="l00360"></a><span class="lineno">  360</span>            nodeRosterReverse[node] = node_name;</div>
<div class="line"><a id="l00361" name="l00361"></a><span class="lineno">  361</span>            nodesRef++;</div>
<div class="line"><a id="l00362" name="l00362"></a><span class="lineno">  362</span>        }</div>
<div class="line"><a id="l00363" name="l00363"></a><span class="lineno">  363</span>        <span class="keywordflow">else</span> {</div>
<div class="line"><a id="l00364" name="l00364"></a><span class="lineno">  364</span>            nodeRoster[name] = node;</div>
<div class="line"><a id="l00365" name="l00365"></a><span class="lineno">  365</span>            nodeRosterReverse[node] = name;</div>
<div class="line"><a id="l00366" name="l00366"></a><span class="lineno">  366</span>        }</div>
<div class="line"><a id="l00367" name="l00367"></a><span class="lineno">  367</span>        <span class="keywordflow">return</span> node;</div>
<div class="line"><a id="l00368" name="l00368"></a><span class="lineno">  368</span>    }</div>
</div>
<div class="line"><a id="l00369" name="l00369"></a><span class="lineno">  369</span> </div>
<div class="foldopen" id="foldopen00370" data-start="{" data-end="}">
<div class="line"><a id="l00370" name="l00370"></a><span class="lineno"><a class="line" href="classnz_1_1graph_1_1_compute_graph.html#a3ef35c2707abf965e7badf00850fb747">  370</a></span>    <a class="code hl_class" href="classnz_1_1nodes_1_1io_1_1_input_node.html">InputNode</a>* <a class="code hl_function" href="classnz_1_1graph_1_1_compute_graph.html#a75a4b72c93448c8617a7af41cc471897">ComputeGraph::addInput</a>(<span class="keyword">const</span> <a class="code hl_class" href="classnz_1_1data_1_1_dimension.html">Tensor::shape_type</a>&amp; shape,</div>
<div class="line"><a id="l00371" name="l00371"></a><span class="lineno">  371</span>        <span class="keyword">const</span> std::initializer_list&lt;Tensor::value_type&gt;&amp; data, <span class="keyword">const</span> <span class="keywordtype">bool</span> requires_grad,</div>
<div class="line"><a id="l00372" name="l00372"></a><span class="lineno">  372</span>        <span class="keyword">const</span> std::string&amp; name) {</div>
<div class="line"><a id="l00373" name="l00373"></a><span class="lineno">  373</span>        <span class="keyword">const</span> <span class="keyword">auto</span> node = <span class="keyword">new</span> <a class="code hl_class" href="classnz_1_1nodes_1_1io_1_1_input_node.html">InputNode</a>(shape, data, requires_grad);</div>
<div class="line"><a id="l00374" name="l00374"></a><span class="lineno">  374</span>        nodes.push_back(node);</div>
<div class="line"><a id="l00375" name="l00375"></a><span class="lineno">  375</span>        inputNodes.push_back(node);</div>
<div class="line"><a id="l00376" name="l00376"></a><span class="lineno">  376</span>        <span class="keywordflow">if</span> (name == <span class="stringliteral">&quot;default&quot;</span>) {</div>
<div class="line"><a id="l00377" name="l00377"></a><span class="lineno">  377</span>            <span class="keyword">const</span> std::string node_name = node-&gt;type + <span class="stringliteral">&quot;_&quot;</span> + std::to_string(nodesRef);</div>
<div class="line"><a id="l00378" name="l00378"></a><span class="lineno">  378</span>            nodeRoster[node_name] = node;</div>
<div class="line"><a id="l00379" name="l00379"></a><span class="lineno">  379</span>            nodeRosterReverse[node] = node_name;</div>
<div class="line"><a id="l00380" name="l00380"></a><span class="lineno">  380</span>            nodesRef++;</div>
<div class="line"><a id="l00381" name="l00381"></a><span class="lineno">  381</span>        }</div>
<div class="line"><a id="l00382" name="l00382"></a><span class="lineno">  382</span>        <span class="keywordflow">else</span> {</div>
<div class="line"><a id="l00383" name="l00383"></a><span class="lineno">  383</span>            nodeRoster[name] = node;</div>
<div class="line"><a id="l00384" name="l00384"></a><span class="lineno">  384</span>            nodeRosterReverse[node] = name;</div>
<div class="line"><a id="l00385" name="l00385"></a><span class="lineno">  385</span>        }</div>
<div class="line"><a id="l00386" name="l00386"></a><span class="lineno">  386</span>        <span class="keywordflow">return</span> node;</div>
<div class="line"><a id="l00387" name="l00387"></a><span class="lineno">  387</span>    }</div>
</div>
<div class="line"><a id="l00388" name="l00388"></a><span class="lineno">  388</span> </div>
<div class="foldopen" id="foldopen00389" data-start="{" data-end="}">
<div class="line"><a id="l00389" name="l00389"></a><span class="lineno"><a class="line" href="classnz_1_1graph_1_1_compute_graph.html#a47cde0bdd9c961d55bfd16c9adb8810e">  389</a></span>    <a class="code hl_class" href="classnz_1_1nodes_1_1io_1_1_output_node.html">OutputNode</a>* <a class="code hl_function" href="classnz_1_1graph_1_1_compute_graph.html#a47cde0bdd9c961d55bfd16c9adb8810e">ComputeGraph::addOutput</a>(<a class="code hl_class" href="classnz_1_1nodes_1_1io_1_1_output_node.html">OutputNode</a>* node, <span class="keyword">const</span> std::string&amp; name) {</div>
<div class="line"><a id="l00390" name="l00390"></a><span class="lineno">  390</span>        nodes.push_back(node);</div>
<div class="line"><a id="l00391" name="l00391"></a><span class="lineno">  391</span>        outputNodes.push_back(node);</div>
<div class="line"><a id="l00392" name="l00392"></a><span class="lineno">  392</span>        <span class="keywordflow">if</span> (name == <span class="stringliteral">&quot;default&quot;</span>) {</div>
<div class="line"><a id="l00393" name="l00393"></a><span class="lineno">  393</span>            <span class="keyword">const</span> std::string node_name = node-&gt;type + <span class="stringliteral">&quot;_&quot;</span> + std::to_string(nodesRef);</div>
<div class="line"><a id="l00394" name="l00394"></a><span class="lineno">  394</span>            nodeRoster[node_name] = node;</div>
<div class="line"><a id="l00395" name="l00395"></a><span class="lineno">  395</span>            nodeRosterReverse[node] = node_name;</div>
<div class="line"><a id="l00396" name="l00396"></a><span class="lineno">  396</span>            nodesRef++;</div>
<div class="line"><a id="l00397" name="l00397"></a><span class="lineno">  397</span>        }</div>
<div class="line"><a id="l00398" name="l00398"></a><span class="lineno">  398</span>        <span class="keywordflow">else</span> {</div>
<div class="line"><a id="l00399" name="l00399"></a><span class="lineno">  399</span>            nodeRoster[name] = node;</div>
<div class="line"><a id="l00400" name="l00400"></a><span class="lineno">  400</span>            nodeRosterReverse[node] = name;</div>
<div class="line"><a id="l00401" name="l00401"></a><span class="lineno">  401</span>        }</div>
<div class="line"><a id="l00402" name="l00402"></a><span class="lineno">  402</span>        <span class="keywordflow">return</span> node;</div>
<div class="line"><a id="l00403" name="l00403"></a><span class="lineno">  403</span>    }</div>
</div>
<div class="line"><a id="l00404" name="l00404"></a><span class="lineno">  404</span> </div>
<div class="foldopen" id="foldopen00405" data-start="{" data-end="}">
<div class="line"><a id="l00405" name="l00405"></a><span class="lineno"><a class="line" href="classnz_1_1graph_1_1_compute_graph.html#a0c5a596f38e6d7e8c1c78e19b9b56167">  405</a></span>    <span class="keywordtype">void</span> <a class="code hl_function" href="classnz_1_1graph_1_1_compute_graph.html#a0c5a596f38e6d7e8c1c78e19b9b56167">ComputeGraph::forward</a>() {</div>
<div class="line"><a id="l00406" name="l00406"></a><span class="lineno">  406</span>        <span class="keywordflow">if</span> (!<a class="code hl_function" href="classnz_1_1graph_1_1_compute_graph.html#aede9d05b6c8e7394ea730b5ffea42164">isSorted</a>()) {</div>
<div class="line"><a id="l00407" name="l00407"></a><span class="lineno">  407</span>            <a class="code hl_function" href="classnz_1_1graph_1_1_compute_graph.html#a5960e75d631200994c0a2f78f58674dd">topologicalSort</a>();</div>
<div class="line"><a id="l00408" name="l00408"></a><span class="lineno">  408</span>        }</div>
<div class="line"><a id="l00409" name="l00409"></a><span class="lineno">  409</span>        <span class="keywordflow">for</span> (<a class="code hl_class" href="classnz_1_1nodes_1_1_node.html">Node</a>* node : sortedNodes) {</div>
<div class="line"><a id="l00410" name="l00410"></a><span class="lineno">  410</span>            node-&gt;forward();</div>
<div class="line"><a id="l00411" name="l00411"></a><span class="lineno">  411</span>        }</div>
<div class="line"><a id="l00412" name="l00412"></a><span class="lineno">  412</span>    }</div>
</div>
<div class="line"><a id="l00413" name="l00413"></a><span class="lineno">  413</span> </div>
<div class="foldopen" id="foldopen00414" data-start="{" data-end="}">
<div class="line"><a id="l00414" name="l00414"></a><span class="lineno"><a class="line" href="classnz_1_1graph_1_1_compute_graph.html#adfcad402fe85084ddbb4f8b6415adc05">  414</a></span>    <span class="keywordtype">void</span> <a class="code hl_function" href="classnz_1_1graph_1_1_compute_graph.html#adfcad402fe85084ddbb4f8b6415adc05">ComputeGraph::backward</a>() {</div>
<div class="line"><a id="l00415" name="l00415"></a><span class="lineno">  415</span>        <span class="keywordflow">if</span> (!<a class="code hl_function" href="classnz_1_1graph_1_1_compute_graph.html#aede9d05b6c8e7394ea730b5ffea42164">isSorted</a>()) {</div>
<div class="line"><a id="l00416" name="l00416"></a><span class="lineno">  416</span>            <span class="keywordflow">throw</span> std::runtime_error(<span class="stringliteral">&quot;Graph is not sorted&quot;</span>);</div>
<div class="line"><a id="l00417" name="l00417"></a><span class="lineno">  417</span>        }</div>
<div class="line"><a id="l00418" name="l00418"></a><span class="lineno">  418</span>        <span class="keywordflow">if</span> (outputNodes.size() == 1) {</div>
<div class="line"><a id="l00419" name="l00419"></a><span class="lineno">  419</span>            outputNodes[0]-&gt;output-&gt;sync();</div>
<div class="line"><a id="l00420" name="l00420"></a><span class="lineno">  420</span>            <span class="keywordflow">for</span> (<span class="keyword">auto</span> it = sortedNodes.rbegin(); it != sortedNodes.rend(); ++it) {</div>
<div class="line"><a id="l00421" name="l00421"></a><span class="lineno">  421</span>                (*it)-&gt;backward();</div>
<div class="line"><a id="l00422" name="l00422"></a><span class="lineno">  422</span>            }</div>
<div class="line"><a id="l00423" name="l00423"></a><span class="lineno">  423</span>        }</div>
<div class="line"><a id="l00424" name="l00424"></a><span class="lineno">  424</span>        <span class="keywordflow">else</span> {</div>
<div class="line"><a id="l00425" name="l00425"></a><span class="lineno">  425</span>            <span class="keywordflow">if</span> (outputNodes.empty()) {</div>
<div class="line"><a id="l00426" name="l00426"></a><span class="lineno">  426</span>                <span class="keywordflow">throw</span> std::runtime_error(<span class="stringliteral">&quot;No output node&quot;</span>);</div>
<div class="line"><a id="l00427" name="l00427"></a><span class="lineno">  427</span>            }</div>
<div class="line"><a id="l00428" name="l00428"></a><span class="lineno">  428</span>            <span class="keywordflow">else</span> {</div>
<div class="line"><a id="l00429" name="l00429"></a><span class="lineno">  429</span>                <span class="keywordflow">throw</span> std::runtime_error(<span class="stringliteral">&quot;Multiple output nodes&quot;</span>);</div>
<div class="line"><a id="l00430" name="l00430"></a><span class="lineno">  430</span>            }</div>
<div class="line"><a id="l00431" name="l00431"></a><span class="lineno">  431</span>        }</div>
<div class="line"><a id="l00432" name="l00432"></a><span class="lineno">  432</span>    }</div>
</div>
<div class="line"><a id="l00433" name="l00433"></a><span class="lineno">  433</span> </div>
<div class="foldopen" id="foldopen00434" data-start="{" data-end="}">
<div class="line"><a id="l00434" name="l00434"></a><span class="lineno"><a class="line" href="classnz_1_1graph_1_1_compute_graph.html#a3f94019acc205a47726c0b0797b0e631">  434</a></span>    <span class="keywordtype">void</span> <a class="code hl_function" href="classnz_1_1graph_1_1_compute_graph.html#a3f94019acc205a47726c0b0797b0e631">ComputeGraph::zeroGrad</a>()<span class="keyword"> const </span>{</div>
<div class="line"><a id="l00435" name="l00435"></a><span class="lineno">  435</span>        <span class="keywordflow">for</span> (<a class="code hl_class" href="classnz_1_1nodes_1_1_node.html">Node</a>* node : nodes) {</div>
<div class="line"><a id="l00436" name="l00436"></a><span class="lineno">  436</span>            node-&gt;output-&gt;zeroGrad();</div>
<div class="line"><a id="l00437" name="l00437"></a><span class="lineno">  437</span>        }</div>
<div class="line"><a id="l00438" name="l00438"></a><span class="lineno">  438</span>    }</div>
</div>
<div class="line"><a id="l00439" name="l00439"></a><span class="lineno">  439</span> </div>
<div class="foldopen" id="foldopen00440" data-start="{" data-end="}">
<div class="line"><a id="l00440" name="l00440"></a><span class="lineno"><a class="line" href="classnz_1_1graph_1_1_compute_graph.html#a856ffb66fa42357ab1a12f9f74d2aba4">  440</a></span>    <span class="keywordtype">void</span> <a class="code hl_function" href="classnz_1_1graph_1_1_compute_graph.html#a856ffb66fa42357ab1a12f9f74d2aba4">ComputeGraph::randomize</a>(<span class="keyword">const</span> std::string&amp; name, <span class="keywordtype">unsigned</span> <span class="keywordtype">long</span> <span class="keywordtype">long</span> seed) {</div>
<div class="line"><a id="l00441" name="l00441"></a><span class="lineno">  441</span>        <span class="keywordflow">if</span> (nodeRoster.find(name) != nodeRoster.end()) {</div>
<div class="line"><a id="l00442" name="l00442"></a><span class="lineno">  442</span>            nodeRoster[name]-&gt;output-&gt;randomize(seed);</div>
<div class="line"><a id="l00443" name="l00443"></a><span class="lineno">  443</span>        }</div>
<div class="line"><a id="l00444" name="l00444"></a><span class="lineno">  444</span>        <span class="keywordflow">else</span> {</div>
<div class="line"><a id="l00445" name="l00445"></a><span class="lineno">  445</span>            <span class="keywordflow">throw</span> std::runtime_error(<span class="stringliteral">&quot;Node not in graph&quot;</span>);</div>
<div class="line"><a id="l00446" name="l00446"></a><span class="lineno">  446</span>        }</div>
<div class="line"><a id="l00447" name="l00447"></a><span class="lineno">  447</span>    }</div>
</div>
<div class="line"><a id="l00448" name="l00448"></a><span class="lineno">  448</span> </div>
<div class="foldopen" id="foldopen00449" data-start="{" data-end="}">
<div class="line"><a id="l00449" name="l00449"></a><span class="lineno"><a class="line" href="classnz_1_1graph_1_1_compute_graph.html#a8d767679a2ff3fcb75a863e7ff808170">  449</a></span>    <span class="keywordtype">void</span> <a class="code hl_function" href="classnz_1_1graph_1_1_compute_graph.html#a856ffb66fa42357ab1a12f9f74d2aba4">ComputeGraph::randomize</a>(<span class="keyword">const</span> <a class="code hl_class" href="classnz_1_1nodes_1_1_node.html">Node</a>* node, <span class="keywordtype">unsigned</span> <span class="keywordtype">long</span> <span class="keywordtype">long</span> seed) {</div>
<div class="line"><a id="l00450" name="l00450"></a><span class="lineno">  450</span>        <span class="keywordflow">if</span> (std::find(nodes.begin(), nodes.end(), node) != nodes.end()) {</div>
<div class="line"><a id="l00451" name="l00451"></a><span class="lineno">  451</span>            node-&gt;output-&gt;randomize(seed);</div>
<div class="line"><a id="l00452" name="l00452"></a><span class="lineno">  452</span>        }</div>
<div class="line"><a id="l00453" name="l00453"></a><span class="lineno">  453</span>        <span class="keywordflow">else</span> {</div>
<div class="line"><a id="l00454" name="l00454"></a><span class="lineno">  454</span>            <span class="keywordflow">throw</span> std::runtime_error(<span class="stringliteral">&quot;Node not in graph&quot;</span>);</div>
<div class="line"><a id="l00455" name="l00455"></a><span class="lineno">  455</span>        }</div>
<div class="line"><a id="l00456" name="l00456"></a><span class="lineno">  456</span>    }</div>
</div>
<div class="line"><a id="l00457" name="l00457"></a><span class="lineno">  457</span> </div>
<div class="foldopen" id="foldopen00458" data-start="{" data-end="}">
<div class="line"><a id="l00458" name="l00458"></a><span class="lineno"><a class="line" href="classnz_1_1graph_1_1_compute_graph.html#a90e4737107c4ec96a1a95022daa6de2f">  458</a></span>    <span class="keywordtype">void</span> <a class="code hl_function" href="classnz_1_1graph_1_1_compute_graph.html#a90e4737107c4ec96a1a95022daa6de2f">ComputeGraph::randomizeAll</a>()<span class="keyword"> const </span>{</div>
<div class="line"><a id="l00459" name="l00459"></a><span class="lineno">  459</span>        <span class="keywordtype">unsigned</span> <span class="keywordtype">long</span> <span class="keywordtype">long</span> seed = std::chrono::system_clock::now().time_since_epoch().count();</div>
<div class="line"><a id="l00460" name="l00460"></a><span class="lineno">  460</span>        <span class="keywordflow">for</span> (<a class="code hl_class" href="classnz_1_1nodes_1_1_node.html">Node</a>* node : inputNodes) {</div>
<div class="line"><a id="l00461" name="l00461"></a><span class="lineno">  461</span>            node-&gt;output-&gt;randomize();</div>
<div class="line"><a id="l00462" name="l00462"></a><span class="lineno">  462</span>            seed++;</div>
<div class="line"><a id="l00463" name="l00463"></a><span class="lineno">  463</span>        }</div>
<div class="line"><a id="l00464" name="l00464"></a><span class="lineno">  464</span>    }</div>
</div>
<div class="line"><a id="l00465" name="l00465"></a><span class="lineno">  465</span> </div>
<div class="foldopen" id="foldopen00466" data-start="{" data-end="}">
<div class="line"><a id="l00466" name="l00466"></a><span class="lineno"><a class="line" href="classnz_1_1graph_1_1_compute_graph.html#ae732145241d1da547219a3e8b95a04ee">  466</a></span>    <span class="keywordtype">void</span> <a class="code hl_function" href="classnz_1_1graph_1_1_compute_graph.html#ae732145241d1da547219a3e8b95a04ee">ComputeGraph::fill</a>(<span class="keyword">const</span> std::string&amp; name, <span class="keyword">const</span> Tensor::value_type val) {</div>
<div class="line"><a id="l00467" name="l00467"></a><span class="lineno">  467</span>        <span class="keywordflow">if</span> (nodeRoster.find(name) != nodeRoster.end()) {</div>
<div class="line"><a id="l00468" name="l00468"></a><span class="lineno">  468</span>            nodeRoster[name]-&gt;output-&gt;fill(val);</div>
<div class="line"><a id="l00469" name="l00469"></a><span class="lineno">  469</span>        }</div>
<div class="line"><a id="l00470" name="l00470"></a><span class="lineno">  470</span>        <span class="keywordflow">else</span> {</div>
<div class="line"><a id="l00471" name="l00471"></a><span class="lineno">  471</span>            <span class="keywordflow">throw</span> std::runtime_error(<span class="stringliteral">&quot;Node not in graph&quot;</span>);</div>
<div class="line"><a id="l00472" name="l00472"></a><span class="lineno">  472</span>        }</div>
<div class="line"><a id="l00473" name="l00473"></a><span class="lineno">  473</span>    }</div>
</div>
<div class="line"><a id="l00474" name="l00474"></a><span class="lineno">  474</span> </div>
<div class="foldopen" id="foldopen00475" data-start="{" data-end="}">
<div class="line"><a id="l00475" name="l00475"></a><span class="lineno"><a class="line" href="classnz_1_1graph_1_1_compute_graph.html#a0f7e2b25f6b75998e0e912320607f6e8">  475</a></span>    <span class="keywordtype">void</span> <a class="code hl_function" href="classnz_1_1graph_1_1_compute_graph.html#ae732145241d1da547219a3e8b95a04ee">ComputeGraph::fill</a>(<span class="keyword">const</span> <a class="code hl_class" href="classnz_1_1nodes_1_1_node.html">Node</a>* node, <span class="keyword">const</span> Tensor::value_type val) {</div>
<div class="line"><a id="l00476" name="l00476"></a><span class="lineno">  476</span>        <span class="keywordflow">if</span> (std::find(nodes.begin(), nodes.end(), node) != nodes.end()) {</div>
<div class="line"><a id="l00477" name="l00477"></a><span class="lineno">  477</span>            node-&gt;output-&gt;fill(val);</div>
<div class="line"><a id="l00478" name="l00478"></a><span class="lineno">  478</span>        }</div>
<div class="line"><a id="l00479" name="l00479"></a><span class="lineno">  479</span>        <span class="keywordflow">else</span> {</div>
<div class="line"><a id="l00480" name="l00480"></a><span class="lineno">  480</span>            <span class="keywordflow">throw</span> std::runtime_error(<span class="stringliteral">&quot;Node not in graph&quot;</span>);</div>
<div class="line"><a id="l00481" name="l00481"></a><span class="lineno">  481</span>        }</div>
<div class="line"><a id="l00482" name="l00482"></a><span class="lineno">  482</span>    }</div>
</div>
<div class="line"><a id="l00483" name="l00483"></a><span class="lineno">  483</span> </div>
<div class="foldopen" id="foldopen00484" data-start="{" data-end="}">
<div class="line"><a id="l00484" name="l00484"></a><span class="lineno"><a class="line" href="classnz_1_1graph_1_1_compute_graph.html#adf21c05c4515016d34d5fc7d639b9e40">  484</a></span>    <span class="keywordtype">void</span> <a class="code hl_function" href="classnz_1_1graph_1_1_compute_graph.html#adf21c05c4515016d34d5fc7d639b9e40">ComputeGraph::fillAll</a>(Tensor::value_type val)<span class="keyword"> const </span>{</div>
<div class="line"><a id="l00485" name="l00485"></a><span class="lineno">  485</span>        <span class="keywordflow">for</span> (<a class="code hl_class" href="classnz_1_1nodes_1_1_node.html">Node</a>* node : inputNodes) {</div>
<div class="line"><a id="l00486" name="l00486"></a><span class="lineno">  486</span>            node-&gt;output-&gt;fill(val);</div>
<div class="line"><a id="l00487" name="l00487"></a><span class="lineno">  487</span>        }</div>
<div class="line"><a id="l00488" name="l00488"></a><span class="lineno">  488</span>    }</div>
</div>
<div class="line"><a id="l00489" name="l00489"></a><span class="lineno">  489</span> </div>
<div class="foldopen" id="foldopen00490" data-start="{" data-end="}">
<div class="line"><a id="l00490" name="l00490"></a><span class="lineno"><a class="line" href="classnz_1_1graph_1_1_compute_graph.html#a8b3498c50429d631b07b5906e7455614">  490</a></span>    <span class="keywordtype">void</span> <a class="code hl_function" href="classnz_1_1graph_1_1_compute_graph.html#a8b3498c50429d631b07b5906e7455614">ComputeGraph::setInput</a>(<span class="keyword">const</span> std::string&amp; name, Tensor::value_type* data) {</div>
<div class="line"><a id="l00491" name="l00491"></a><span class="lineno">  491</span>        <span class="keywordflow">if</span> (nodeRoster.find(name) != nodeRoster.end()) {</div>
<div class="line"><a id="l00492" name="l00492"></a><span class="lineno">  492</span>            nodeRoster[name]-&gt;output-&gt;dataInject(data);</div>
<div class="line"><a id="l00493" name="l00493"></a><span class="lineno">  493</span>        }</div>
<div class="line"><a id="l00494" name="l00494"></a><span class="lineno">  494</span>        <span class="keywordflow">else</span> {</div>
<div class="line"><a id="l00495" name="l00495"></a><span class="lineno">  495</span>            <span class="keywordflow">throw</span> std::runtime_error(<span class="stringliteral">&quot;Node not in graph&quot;</span>);</div>
<div class="line"><a id="l00496" name="l00496"></a><span class="lineno">  496</span>        }</div>
<div class="line"><a id="l00497" name="l00497"></a><span class="lineno">  497</span>    }</div>
</div>
<div class="line"><a id="l00498" name="l00498"></a><span class="lineno">  498</span> </div>
<div class="foldopen" id="foldopen00499" data-start="{" data-end="}">
<div class="line"><a id="l00499" name="l00499"></a><span class="lineno"><a class="line" href="classnz_1_1graph_1_1_compute_graph.html#aa32640d8edb36f42e6093cd70037b5f3">  499</a></span>    <span class="keywordtype">void</span> <a class="code hl_function" href="classnz_1_1graph_1_1_compute_graph.html#a8b3498c50429d631b07b5906e7455614">ComputeGraph::setInput</a>(<span class="keyword">const</span> <a class="code hl_class" href="classnz_1_1nodes_1_1_node.html">Node</a>* node, Tensor::value_type* data) {</div>
<div class="line"><a id="l00500" name="l00500"></a><span class="lineno">  500</span>        <span class="keywordflow">if</span> (std::find(nodes.begin(), nodes.end(), node) != nodes.end()) {</div>
<div class="line"><a id="l00501" name="l00501"></a><span class="lineno">  501</span>            node-&gt;output-&gt;dataInject(data);</div>
<div class="line"><a id="l00502" name="l00502"></a><span class="lineno">  502</span>        }</div>
<div class="line"><a id="l00503" name="l00503"></a><span class="lineno">  503</span>        <span class="keywordflow">else</span> {</div>
<div class="line"><a id="l00504" name="l00504"></a><span class="lineno">  504</span>            <span class="keywordflow">throw</span> std::runtime_error(<span class="stringliteral">&quot;Node not in graph&quot;</span>);</div>
<div class="line"><a id="l00505" name="l00505"></a><span class="lineno">  505</span>        }</div>
<div class="line"><a id="l00506" name="l00506"></a><span class="lineno">  506</span>    }</div>
</div>
<div class="line"><a id="l00507" name="l00507"></a><span class="lineno">  507</span> </div>
<div class="foldopen" id="foldopen00508" data-start="{" data-end="}">
<div class="line"><a id="l00508" name="l00508"></a><span class="lineno"><a class="line" href="classnz_1_1graph_1_1_compute_graph.html#a75f918fb756d5f7155ee49092ce713e1">  508</a></span>    Tensor::value_type* <a class="code hl_function" href="classnz_1_1graph_1_1_compute_graph.html#a75f918fb756d5f7155ee49092ce713e1">ComputeGraph::getOutput</a>()<span class="keyword"> const </span>{</div>
<div class="line"><a id="l00509" name="l00509"></a><span class="lineno">  509</span>        <span class="keywordflow">if</span> (outputNodes.empty()) {</div>
<div class="line"><a id="l00510" name="l00510"></a><span class="lineno">  510</span>            <span class="keywordflow">throw</span> std::runtime_error(<span class="stringliteral">&quot;No output node&quot;</span>);</div>
<div class="line"><a id="l00511" name="l00511"></a><span class="lineno">  511</span>        }</div>
<div class="line"><a id="l00512" name="l00512"></a><span class="lineno">  512</span>        <span class="keywordflow">return</span> outputNodes[0]-&gt;output-&gt;data();</div>
<div class="line"><a id="l00513" name="l00513"></a><span class="lineno">  513</span>    }</div>
</div>
<div class="line"><a id="l00514" name="l00514"></a><span class="lineno">  514</span> </div>
<div class="foldopen" id="foldopen00515" data-start="{" data-end="}">
<div class="line"><a id="l00515" name="l00515"></a><span class="lineno"><a class="line" href="classnz_1_1graph_1_1_compute_graph.html#aef02d3290fb92b142be346f2fdb125c5">  515</a></span>    Tensor::value_type* <a class="code hl_function" href="classnz_1_1graph_1_1_compute_graph.html#aef02d3290fb92b142be346f2fdb125c5">ComputeGraph::getOutputHost</a>()<span class="keyword"> const </span>{</div>
<div class="line"><a id="l00516" name="l00516"></a><span class="lineno">  516</span>        <span class="keywordflow">if</span> (outputNodes.empty()) {</div>
<div class="line"><a id="l00517" name="l00517"></a><span class="lineno">  517</span>            <span class="keywordflow">throw</span> std::runtime_error(<span class="stringliteral">&quot;No output node&quot;</span>);</div>
<div class="line"><a id="l00518" name="l00518"></a><span class="lineno">  518</span>        }</div>
<div class="line"><a id="l00519" name="l00519"></a><span class="lineno">  519</span>        <span class="keyword">auto</span>* data = <span class="keyword">static_cast&lt;</span>Tensor::value_type*<span class="keyword">&gt;</span>(malloc(</div>
<div class="line"><a id="l00520" name="l00520"></a><span class="lineno">  520</span>            outputNodes[0]-&gt;output-&gt;size() * <span class="keyword">sizeof</span>(Tensor::value_type)));</div>
<div class="line"><a id="l00521" name="l00521"></a><span class="lineno">  521</span>        <a class="code hl_function" href="classnz_1_1cu_strm_1_1_stream_manager.html#ab4b2eb422e0e1ee44bdfdc0eb94457ce">cuStrm::StreamManager&lt;Tensor::value_type&gt;::Instance</a>().<a class="code hl_function" href="classnz_1_1cu_strm_1_1_stream_manager.html#afa38d5c6db0e6b48c8f74ce8ad0df2bc">memcpy</a>(data, outputNodes[0]-&gt;output-&gt;data(), outputNodes[0]-&gt;output-&gt;size() * <span class="keyword">sizeof</span>(Tensor::value_type),</div>
<div class="line"><a id="l00522" name="l00522"></a><span class="lineno">  522</span>                   cudaMemcpyDeviceToHost);</div>
<div class="line"><a id="l00523" name="l00523"></a><span class="lineno">  523</span>        <a class="code hl_function" href="classnz_1_1cu_strm_1_1_stream_manager.html#ab4b2eb422e0e1ee44bdfdc0eb94457ce">cuStrm::StreamManager&lt;Tensor::value_type&gt;::Instance</a>().<a class="code hl_function" href="classnz_1_1cu_strm_1_1_stream_manager.html#abe439fa00c0bd369c0b2345b095ed5af">syncData</a>(data);</div>
<div class="line"><a id="l00524" name="l00524"></a><span class="lineno">  524</span>        <span class="keywordflow">return</span> data;</div>
<div class="line"><a id="l00525" name="l00525"></a><span class="lineno">  525</span>    }</div>
</div>
<div class="line"><a id="l00526" name="l00526"></a><span class="lineno">  526</span> </div>
<div class="foldopen" id="foldopen00527" data-start="{" data-end="}">
<div class="line"><a id="l00527" name="l00527"></a><span class="lineno"><a class="line" href="classnz_1_1graph_1_1_compute_graph.html#a69d293299714e489ce8cc2dba1e5a9a5">  527</a></span>    <a class="code hl_class" href="classnz_1_1nodes_1_1io_1_1_output_node.html">OutputNode</a>* <a class="code hl_function" href="classnz_1_1graph_1_1_compute_graph.html#a69d293299714e489ce8cc2dba1e5a9a5">ComputeGraph::getOutputNode</a>()<span class="keyword"> const </span>{</div>
<div class="line"><a id="l00528" name="l00528"></a><span class="lineno">  528</span>        <span class="keywordflow">if</span> (outputNodes.empty()) {</div>
<div class="line"><a id="l00529" name="l00529"></a><span class="lineno">  529</span>            <span class="keywordflow">throw</span> std::runtime_error(<span class="stringliteral">&quot;No output node&quot;</span>);</div>
<div class="line"><a id="l00530" name="l00530"></a><span class="lineno">  530</span>        }</div>
<div class="line"><a id="l00531" name="l00531"></a><span class="lineno">  531</span>        <span class="keywordflow">return</span> outputNodes[0];</div>
<div class="line"><a id="l00532" name="l00532"></a><span class="lineno">  532</span>    }</div>
</div>
<div class="line"><a id="l00533" name="l00533"></a><span class="lineno">  533</span> </div>
<div class="foldopen" id="foldopen00534" data-start="{" data-end="}">
<div class="line"><a id="l00534" name="l00534"></a><span class="lineno"><a class="line" href="classnz_1_1graph_1_1_compute_graph.html#a3a53a15c04e85da4e6aa0a209fbee4c2">  534</a></span>    Tensor::value_type <a class="code hl_function" href="classnz_1_1graph_1_1_compute_graph.html#a3a53a15c04e85da4e6aa0a209fbee4c2">ComputeGraph::getLoss</a>()<span class="keyword"> const </span>{</div>
<div class="line"><a id="l00535" name="l00535"></a><span class="lineno">  535</span>        <span class="keywordflow">if</span> (outputNodes.empty()) {</div>
<div class="line"><a id="l00536" name="l00536"></a><span class="lineno">  536</span>            <span class="keywordflow">throw</span> std::runtime_error(<span class="stringliteral">&quot;No output node&quot;</span>);</div>
<div class="line"><a id="l00537" name="l00537"></a><span class="lineno">  537</span>        }</div>
<div class="line"><a id="l00538" name="l00538"></a><span class="lineno">  538</span>        <span class="keywordflow">return</span> outputNodes[0]-&gt;getLoss();</div>
<div class="line"><a id="l00539" name="l00539"></a><span class="lineno">  539</span>    }</div>
</div>
<div class="line"><a id="l00540" name="l00540"></a><span class="lineno">  540</span> </div>
<div class="foldopen" id="foldopen00541" data-start="{" data-end="}">
<div class="line"><a id="l00541" name="l00541"></a><span class="lineno"><a class="line" href="classnz_1_1graph_1_1_compute_graph.html#aee306895a3078a0b09d2a4bdf4843200">  541</a></span>    <span class="keywordtype">void</span> <a class="code hl_function" href="classnz_1_1graph_1_1_compute_graph.html#aee306895a3078a0b09d2a4bdf4843200">ComputeGraph::update</a>(<a class="code hl_class" href="classnz_1_1opt_1_1_optimizer.html">Optimizer</a>* optimizer)<span class="keyword"> const </span>{</div>
<div class="line"><a id="l00542" name="l00542"></a><span class="lineno">  542</span>        <span class="keywordflow">for</span> (<a class="code hl_class" href="classnz_1_1nodes_1_1_node.html">Node</a>* node : nodes) {</div>
<div class="line"><a id="l00543" name="l00543"></a><span class="lineno">  543</span>            <span class="keywordflow">if</span> (node-&gt;output-&gt;requiresGrad()) {</div>
<div class="line"><a id="l00544" name="l00544"></a><span class="lineno">  544</span>                optimizer-&gt;<a class="code hl_function" href="classnz_1_1opt_1_1_optimizer.html#a826381abaaf29dbebade7cfd38b266e4">step</a>(node);</div>
<div class="line"><a id="l00545" name="l00545"></a><span class="lineno">  545</span>            }</div>
<div class="line"><a id="l00546" name="l00546"></a><span class="lineno">  546</span>        }</div>
<div class="line"><a id="l00547" name="l00547"></a><span class="lineno">  547</span>    }</div>
</div>
<div class="line"><a id="l00548" name="l00548"></a><span class="lineno">  548</span> </div>
<div class="line"><a id="l00549" name="l00549"></a><span class="lineno">  549</span>    <span class="keywordtype">bool</span> ComputeGraph::inGraph(<a class="code hl_class" href="classnz_1_1nodes_1_1_node.html">Node</a>* node)<span class="keyword"> const </span>{</div>
<div class="line"><a id="l00550" name="l00550"></a><span class="lineno">  550</span>        <span class="keywordflow">return</span> std::find(nodes.begin(), nodes.end(), node) != nodes.end();</div>
<div class="line"><a id="l00551" name="l00551"></a><span class="lineno">  551</span>    }</div>
<div class="line"><a id="l00552" name="l00552"></a><span class="lineno">  552</span> </div>
<div class="foldopen" id="foldopen00553" data-start="{" data-end="}">
<div class="line"><a id="l00553" name="l00553"></a><span class="lineno"><a class="line" href="classnz_1_1graph_1_1_compute_graph.html#a6aa07603223ea714d74fc884218cb50f">  553</a></span>    <span class="keywordtype">void</span> <a class="code hl_function" href="classnz_1_1graph_1_1_compute_graph.html#a6aa07603223ea714d74fc884218cb50f">ComputeGraph::save</a>(<span class="keyword">const</span> std::string&amp; path) {</div>
<div class="line"><a id="l00554" name="l00554"></a><span class="lineno">  554</span>        <span class="keywordflow">if</span> (path.empty()) {</div>
<div class="line"><a id="l00555" name="l00555"></a><span class="lineno">  555</span>            <span class="keywordflow">throw</span> std::runtime_error(<span class="stringliteral">&quot;Path cannot be empty&quot;</span>);</div>
<div class="line"><a id="l00556" name="l00556"></a><span class="lineno">  556</span>        }</div>
<div class="line"><a id="l00557" name="l00557"></a><span class="lineno">  557</span>        <span class="keywordflow">if</span> (sortedNodes.empty()) {</div>
<div class="line"><a id="l00558" name="l00558"></a><span class="lineno">  558</span>            <span class="keywordflow">throw</span> std::runtime_error(<span class="stringliteral">&quot;Graph not sorted&quot;</span>);</div>
<div class="line"><a id="l00559" name="l00559"></a><span class="lineno">  559</span>        }</div>
<div class="line"><a id="l00560" name="l00560"></a><span class="lineno">  560</span> </div>
<div class="line"><a id="l00561" name="l00561"></a><span class="lineno">  561</span>        std::ofstream out(path);</div>
<div class="line"><a id="l00562" name="l00562"></a><span class="lineno">  562</span>        <span class="keywordflow">if</span> (!out.is_open()) {</div>
<div class="line"><a id="l00563" name="l00563"></a><span class="lineno">  563</span>            <span class="keywordflow">throw</span> std::runtime_error(<span class="stringliteral">&quot;Failed to open file for writing&quot;</span>);</div>
<div class="line"><a id="l00564" name="l00564"></a><span class="lineno">  564</span>        }</div>
<div class="line"><a id="l00565" name="l00565"></a><span class="lineno">  565</span> </div>
<div class="line"><a id="l00566" name="l00566"></a><span class="lineno">  566</span>        out &lt;&lt; <span class="stringliteral">&quot;[\n&quot;</span>; <span class="comment">// Start the JSON array</span></div>
<div class="line"><a id="l00567" name="l00567"></a><span class="lineno">  567</span> </div>
<div class="line"><a id="l00568" name="l00568"></a><span class="lineno">  568</span>        <span class="keywordflow">for</span> (<span class="keywordtype">size_t</span> i = 0; i &lt; sortedNodes.size(); ++i) {</div>
<div class="line"><a id="l00569" name="l00569"></a><span class="lineno">  569</span>            <a class="code hl_class" href="classnz_1_1nodes_1_1_node.html">Node</a>* node = sortedNodes[i];</div>
<div class="line"><a id="l00570" name="l00570"></a><span class="lineno">  570</span> </div>
<div class="line"><a id="l00571" name="l00571"></a><span class="lineno">  571</span>            out &lt;&lt; <span class="stringliteral">&quot;  {\n&quot;</span>; <span class="comment">// Start a node object</span></div>
<div class="line"><a id="l00572" name="l00572"></a><span class="lineno">  572</span>            out &lt;&lt; R<span class="stringliteral">&quot;(    &quot;type&quot;: &quot;)&quot; &lt;&lt; node-&gt;type &lt;&lt; </span><span class="stringliteral">&quot;\&quot;,\n&quot;</span>;</div>
<div class="line"><a id="l00573" name="l00573"></a><span class="lineno">  573</span>            out &lt;&lt; R<span class="stringliteral">&quot;(    &quot;name&quot;: &quot;)&quot; &lt;&lt; nodeRosterReverse[node] &lt;&lt; </span><span class="stringliteral">&quot;\&quot;,\n&quot;</span>;</div>
<div class="line"><a id="l00574" name="l00574"></a><span class="lineno">  574</span> </div>
<div class="line"><a id="l00575" name="l00575"></a><span class="lineno">  575</span>            <span class="comment">// Pre nodes (inputs)</span></div>
<div class="line"><a id="l00576" name="l00576"></a><span class="lineno">  576</span>            out &lt;&lt; <span class="stringliteral">&quot;    \&quot;pre\&quot;: [&quot;</span>;</div>
<div class="line"><a id="l00577" name="l00577"></a><span class="lineno">  577</span>            <span class="keywordflow">for</span> (<span class="keywordtype">size_t</span> j = 0; j &lt; node-&gt;inputs.size(); ++j) {</div>
<div class="line"><a id="l00578" name="l00578"></a><span class="lineno">  578</span>                <span class="keyword">auto</span> input = node-&gt;inputs[j];</div>
<div class="line"><a id="l00579" name="l00579"></a><span class="lineno">  579</span>                <span class="keyword">auto</span> index = std::distance(sortedNodes.begin(),</div>
<div class="line"><a id="l00580" name="l00580"></a><span class="lineno">  580</span>                                           std::find(sortedNodes.begin(), sortedNodes.end(), input));</div>
<div class="line"><a id="l00581" name="l00581"></a><span class="lineno">  581</span>                out &lt;&lt; index;</div>
<div class="line"><a id="l00582" name="l00582"></a><span class="lineno">  582</span>                <span class="keywordflow">if</span> (j &lt; node-&gt;inputs.size() - 1) {</div>
<div class="line"><a id="l00583" name="l00583"></a><span class="lineno">  583</span>                    out &lt;&lt; <span class="stringliteral">&quot;, &quot;</span>;</div>
<div class="line"><a id="l00584" name="l00584"></a><span class="lineno">  584</span>                }</div>
<div class="line"><a id="l00585" name="l00585"></a><span class="lineno">  585</span>            }</div>
<div class="line"><a id="l00586" name="l00586"></a><span class="lineno">  586</span>            out &lt;&lt; <span class="stringliteral">&quot;],\n&quot;</span>;</div>
<div class="line"><a id="l00587" name="l00587"></a><span class="lineno">  587</span>            out &lt;&lt; <span class="stringliteral">&quot;    \&quot;post\&quot;: [&quot;</span>;</div>
<div class="line"><a id="l00588" name="l00588"></a><span class="lineno">  588</span>            <span class="keywordflow">for</span> (<span class="keywordtype">size_t</span> j = 0; j &lt; adjList[node].size(); ++j) {</div>
<div class="line"><a id="l00589" name="l00589"></a><span class="lineno">  589</span>                <span class="keyword">auto</span> next = adjList[node][j];</div>
<div class="line"><a id="l00590" name="l00590"></a><span class="lineno">  590</span>                <span class="keyword">auto</span> index = std::distance(sortedNodes.begin(),</div>
<div class="line"><a id="l00591" name="l00591"></a><span class="lineno">  591</span>                                           std::find(sortedNodes.begin(), sortedNodes.end(), next));</div>
<div class="line"><a id="l00592" name="l00592"></a><span class="lineno">  592</span>                out &lt;&lt; index;</div>
<div class="line"><a id="l00593" name="l00593"></a><span class="lineno">  593</span>                <span class="keywordflow">if</span> (j &lt; adjList[node].size() - 1) {</div>
<div class="line"><a id="l00594" name="l00594"></a><span class="lineno">  594</span>                    out &lt;&lt; <span class="stringliteral">&quot;, &quot;</span>;</div>
<div class="line"><a id="l00595" name="l00595"></a><span class="lineno">  595</span>                }</div>
<div class="line"><a id="l00596" name="l00596"></a><span class="lineno">  596</span>            }</div>
<div class="line"><a id="l00597" name="l00597"></a><span class="lineno">  597</span>            out &lt;&lt; <span class="stringliteral">&quot;],\n&quot;</span>;</div>
<div class="line"><a id="l00598" name="l00598"></a><span class="lineno">  598</span>            out &lt;&lt; <span class="stringliteral">&quot;    \&quot;shape\&quot;: [&quot;</span></div>
<div class="line"><a id="l00599" name="l00599"></a><span class="lineno">  599</span>                &lt;&lt; node-&gt;output-&gt;shape()[0] &lt;&lt; <span class="stringliteral">&quot;, &quot;</span></div>
<div class="line"><a id="l00600" name="l00600"></a><span class="lineno">  600</span>                &lt;&lt; node-&gt;output-&gt;shape()[1] &lt;&lt; <span class="stringliteral">&quot;],\n&quot;</span>;</div>
<div class="line"><a id="l00601" name="l00601"></a><span class="lineno">  601</span>            out &lt;&lt; <span class="stringliteral">&quot;    \&quot;data\&quot;: [&quot;</span>;</div>
<div class="line"><a id="l00602" name="l00602"></a><span class="lineno">  602</span>            <span class="keyword">auto</span>* data = <span class="keyword">new</span> <span class="keywordtype">float</span>[node-&gt;output-&gt;size()];</div>
<div class="line"><a id="l00603" name="l00603"></a><span class="lineno">  603</span>            <a class="code hl_function" href="classnz_1_1cu_strm_1_1_stream_manager.html#ab4b2eb422e0e1ee44bdfdc0eb94457ce">cuStrm::StreamManager&lt;Tensor::value_type&gt;::Instance</a>().<a class="code hl_function" href="classnz_1_1cu_strm_1_1_stream_manager.html#afa38d5c6db0e6b48c8f74ce8ad0df2bc">memcpy</a>(data, node-&gt;output-&gt;data(), node-&gt;output-&gt;size() * <span class="keyword">sizeof</span>(<span class="keywordtype">float</span>), cudaMemcpyDeviceToHost);</div>
<div class="line"><a id="l00604" name="l00604"></a><span class="lineno">  604</span>            <a class="code hl_function" href="classnz_1_1cu_strm_1_1_stream_manager.html#ab4b2eb422e0e1ee44bdfdc0eb94457ce">cuStrm::StreamManager&lt;Tensor::value_type&gt;::Instance</a>().<a class="code hl_function" href="classnz_1_1cu_strm_1_1_stream_manager.html#abe439fa00c0bd369c0b2345b095ed5af">syncData</a>(data);</div>
<div class="line"><a id="l00605" name="l00605"></a><span class="lineno">  605</span>            <span class="keywordflow">for</span> (<span class="keywordtype">size_t</span> j = 0; j &lt; node-&gt;output-&gt;size(); ++j) {</div>
<div class="line"><a id="l00606" name="l00606"></a><span class="lineno">  606</span>                out &lt;&lt; data[j];</div>
<div class="line"><a id="l00607" name="l00607"></a><span class="lineno">  607</span>                <span class="keywordflow">if</span> (j &lt; node-&gt;output-&gt;size() - 1) {</div>
<div class="line"><a id="l00608" name="l00608"></a><span class="lineno">  608</span>                    out &lt;&lt; <span class="stringliteral">&quot;, &quot;</span>;</div>
<div class="line"><a id="l00609" name="l00609"></a><span class="lineno">  609</span>                }</div>
<div class="line"><a id="l00610" name="l00610"></a><span class="lineno">  610</span>            }</div>
<div class="line"><a id="l00611" name="l00611"></a><span class="lineno">  611</span>            out &lt;&lt; <span class="stringliteral">&quot;],\n&quot;</span>;</div>
<div class="line"><a id="l00612" name="l00612"></a><span class="lineno">  612</span>            <span class="keyword">delete</span>[] data;</div>
<div class="line"><a id="l00613" name="l00613"></a><span class="lineno">  613</span>            out &lt;&lt; <span class="stringliteral">&quot;    \&quot;requires_grad\&quot;: &quot;</span> &lt;&lt; (node-&gt;output-&gt;requiresGrad() ? <span class="stringliteral">&quot;true&quot;</span> : <span class="stringliteral">&quot;false&quot;</span>) &lt;&lt; <span class="stringliteral">&quot;,\n&quot;</span>;</div>
<div class="line"><a id="l00614" name="l00614"></a><span class="lineno">  614</span>            <span class="keywordflow">if</span> (node-&gt;output-&gt;requiresGrad()) {</div>
<div class="line"><a id="l00615" name="l00615"></a><span class="lineno">  615</span>                out &lt;&lt; <span class="stringliteral">&quot;    \&quot;grad\&quot;: [&quot;</span>;</div>
<div class="line"><a id="l00616" name="l00616"></a><span class="lineno">  616</span>                <span class="keyword">auto</span>* grad_data = <span class="keyword">new</span> <span class="keywordtype">float</span>[node-&gt;output-&gt;size()];</div>
<div class="line"><a id="l00617" name="l00617"></a><span class="lineno">  617</span>                <a class="code hl_function" href="classnz_1_1cu_strm_1_1_stream_manager.html#ab4b2eb422e0e1ee44bdfdc0eb94457ce">cuStrm::StreamManager&lt;Tensor::value_type&gt;::Instance</a>().<a class="code hl_function" href="classnz_1_1cu_strm_1_1_stream_manager.html#afa38d5c6db0e6b48c8f74ce8ad0df2bc">memcpy</a>(grad_data, node-&gt;output-&gt;grad(), node-&gt;output-&gt;size() * <span class="keyword">sizeof</span>(<span class="keywordtype">float</span>),</div>
<div class="line"><a id="l00618" name="l00618"></a><span class="lineno">  618</span>                           cudaMemcpyDeviceToHost);</div>
<div class="line"><a id="l00619" name="l00619"></a><span class="lineno">  619</span>                <a class="code hl_function" href="classnz_1_1cu_strm_1_1_stream_manager.html#ab4b2eb422e0e1ee44bdfdc0eb94457ce">cuStrm::StreamManager&lt;Tensor::value_type&gt;::Instance</a>().<a class="code hl_function" href="classnz_1_1cu_strm_1_1_stream_manager.html#abe439fa00c0bd369c0b2345b095ed5af">syncData</a>(grad_data);</div>
<div class="line"><a id="l00620" name="l00620"></a><span class="lineno">  620</span>                <span class="keywordflow">for</span> (<span class="keywordtype">size_t</span> j = 0; j &lt; node-&gt;output-&gt;size(); ++j) {</div>
<div class="line"><a id="l00621" name="l00621"></a><span class="lineno">  621</span>                    out &lt;&lt; grad_data[j];</div>
<div class="line"><a id="l00622" name="l00622"></a><span class="lineno">  622</span>                    <span class="keywordflow">if</span> (j &lt; node-&gt;output-&gt;size() - 1) {</div>
<div class="line"><a id="l00623" name="l00623"></a><span class="lineno">  623</span>                        out &lt;&lt; <span class="stringliteral">&quot;, &quot;</span>;</div>
<div class="line"><a id="l00624" name="l00624"></a><span class="lineno">  624</span>                    }</div>
<div class="line"><a id="l00625" name="l00625"></a><span class="lineno">  625</span>                }</div>
<div class="line"><a id="l00626" name="l00626"></a><span class="lineno">  626</span>                out &lt;&lt; <span class="stringliteral">&quot;]\n&quot;</span>;</div>
<div class="line"><a id="l00627" name="l00627"></a><span class="lineno">  627</span>                <span class="keyword">delete</span>[] grad_data;</div>
<div class="line"><a id="l00628" name="l00628"></a><span class="lineno">  628</span>            }</div>
<div class="line"><a id="l00629" name="l00629"></a><span class="lineno">  629</span>            <span class="keywordflow">else</span> {</div>
<div class="line"><a id="l00630" name="l00630"></a><span class="lineno">  630</span>                out &lt;&lt; <span class="stringliteral">&quot;    \&quot;grad\&quot;: []\n&quot;</span>;</div>
<div class="line"><a id="l00631" name="l00631"></a><span class="lineno">  631</span>            }</div>
<div class="line"><a id="l00632" name="l00632"></a><span class="lineno">  632</span>            out &lt;&lt; <span class="stringliteral">&quot;  }&quot;</span>;</div>
<div class="line"><a id="l00633" name="l00633"></a><span class="lineno">  633</span>            <span class="keywordflow">if</span> (i &lt; sortedNodes.size() - 1) {</div>
<div class="line"><a id="l00634" name="l00634"></a><span class="lineno">  634</span>                out &lt;&lt; <span class="stringliteral">&quot;,&quot;</span>;</div>
<div class="line"><a id="l00635" name="l00635"></a><span class="lineno">  635</span>            }</div>
<div class="line"><a id="l00636" name="l00636"></a><span class="lineno">  636</span>            out &lt;&lt; <span class="stringliteral">&quot;\n&quot;</span>;</div>
<div class="line"><a id="l00637" name="l00637"></a><span class="lineno">  637</span>        }</div>
<div class="line"><a id="l00638" name="l00638"></a><span class="lineno">  638</span>        out &lt;&lt; <span class="stringliteral">&quot;]\n&quot;</span>;</div>
<div class="line"><a id="l00639" name="l00639"></a><span class="lineno">  639</span>        out.close();</div>
<div class="line"><a id="l00640" name="l00640"></a><span class="lineno">  640</span>    }</div>
</div>
<div class="line"><a id="l00641" name="l00641"></a><span class="lineno">  641</span> </div>
<div class="foldopen" id="foldopen00642" data-start="{" data-end="}">
<div class="line"><a id="l00642" name="l00642"></a><span class="lineno"><a class="line" href="classnz_1_1graph_1_1_compute_graph.html#a9748d859d952864bfa498a79c9fb394a">  642</a></span>    <span class="keywordtype">void</span> <a class="code hl_function" href="classnz_1_1graph_1_1_compute_graph.html#a9748d859d952864bfa498a79c9fb394a">ComputeGraph::load</a>(<span class="keyword">const</span> std::string&amp; path) {</div>
<div class="line"><a id="l00643" name="l00643"></a><span class="lineno">  643</span>        <span class="keywordflow">if</span> (path.empty()) {</div>
<div class="line"><a id="l00644" name="l00644"></a><span class="lineno">  644</span>            <span class="keywordflow">throw</span> std::runtime_error(<span class="stringliteral">&quot;Path cannot be empty&quot;</span>);</div>
<div class="line"><a id="l00645" name="l00645"></a><span class="lineno">  645</span>        }</div>
<div class="line"><a id="l00646" name="l00646"></a><span class="lineno">  646</span>        <span class="keywordflow">if</span> (!nodes.empty()) {</div>
<div class="line"><a id="l00647" name="l00647"></a><span class="lineno">  647</span>            <span class="keywordflow">throw</span> std::runtime_error(<span class="stringliteral">&quot;Graph already loaded&quot;</span>);</div>
<div class="line"><a id="l00648" name="l00648"></a><span class="lineno">  648</span>        }</div>
<div class="line"><a id="l00649" name="l00649"></a><span class="lineno">  649</span>        std::ifstream in(path);</div>
<div class="line"><a id="l00650" name="l00650"></a><span class="lineno">  650</span>        <span class="keywordflow">if</span> (!in.is_open()) {</div>
<div class="line"><a id="l00651" name="l00651"></a><span class="lineno">  651</span>            <span class="keywordflow">throw</span> std::runtime_error(<span class="stringliteral">&quot;Failed to open file for reading&quot;</span>);</div>
<div class="line"><a id="l00652" name="l00652"></a><span class="lineno">  652</span>        }</div>
<div class="line"><a id="l00653" name="l00653"></a><span class="lineno">  653</span>        <span class="keywordtype">int</span> i = 0;</div>
<div class="line"><a id="l00654" name="l00654"></a><span class="lineno">  654</span>        std::string line;</div>
<div class="line"><a id="l00655" name="l00655"></a><span class="lineno">  655</span>        std::string type;</div>
<div class="line"><a id="l00656" name="l00656"></a><span class="lineno">  656</span>        std::string name;</div>
<div class="line"><a id="l00657" name="l00657"></a><span class="lineno">  657</span>        std::vector&lt;int&gt; pre;</div>
<div class="line"><a id="l00658" name="l00658"></a><span class="lineno">  658</span>        std::vector&lt;int&gt; post;</div>
<div class="line"><a id="l00659" name="l00659"></a><span class="lineno">  659</span>        <a class="code hl_class" href="classnz_1_1data_1_1_dimension.html">Tensor::shape_type</a> shape;</div>
<div class="line"><a id="l00660" name="l00660"></a><span class="lineno">  660</span>        <span class="keywordtype">float</span>* data = <span class="keyword">nullptr</span>;</div>
<div class="line"><a id="l00661" name="l00661"></a><span class="lineno">  661</span>        <span class="keywordtype">bool</span> requires_grad = <span class="keyword">false</span>;</div>
<div class="line"><a id="l00662" name="l00662"></a><span class="lineno">  662</span>        <span class="keywordtype">float</span>* grad = <span class="keyword">nullptr</span>;</div>
<div class="line"><a id="l00663" name="l00663"></a><span class="lineno">  663</span>        <span class="keywordtype">bool</span> reading_node = <span class="keyword">false</span>;</div>
<div class="line"><a id="l00664" name="l00664"></a><span class="lineno">  664</span>        <span class="keywordflow">while</span> (std::getline(in, line)) {</div>
<div class="line"><a id="l00665" name="l00665"></a><span class="lineno">  665</span>            <span class="keywordflow">if</span> (line.empty()) {</div>
<div class="line"><a id="l00666" name="l00666"></a><span class="lineno">  666</span>                <span class="keywordflow">continue</span>;</div>
<div class="line"><a id="l00667" name="l00667"></a><span class="lineno">  667</span>            }</div>
<div class="line"><a id="l00668" name="l00668"></a><span class="lineno">  668</span>            <span class="keywordflow">if</span> (line == <span class="stringliteral">&quot;[&quot;</span> &amp;&amp; !reading_node) {</div>
<div class="line"><a id="l00669" name="l00669"></a><span class="lineno">  669</span>                <span class="keywordflow">continue</span>;</div>
<div class="line"><a id="l00670" name="l00670"></a><span class="lineno">  670</span>            }</div>
<div class="line"><a id="l00671" name="l00671"></a><span class="lineno">  671</span>            <span class="keywordflow">if</span> (line == <span class="stringliteral">&quot;]&quot;</span> &amp;&amp; !reading_node) {</div>
<div class="line"><a id="l00672" name="l00672"></a><span class="lineno">  672</span>                <span class="keywordflow">break</span>;</div>
<div class="line"><a id="l00673" name="l00673"></a><span class="lineno">  673</span>            }</div>
<div class="line"><a id="l00674" name="l00674"></a><span class="lineno">  674</span>            <span class="keywordflow">if</span> (line.find(<span class="charliteral">&#39;{&#39;</span>) != std::string::npos &amp;&amp; !reading_node) {</div>
<div class="line"><a id="l00675" name="l00675"></a><span class="lineno">  675</span>                reading_node = <span class="keyword">true</span>;</div>
<div class="line"><a id="l00676" name="l00676"></a><span class="lineno">  676</span>                <span class="keywordflow">continue</span>;</div>
<div class="line"><a id="l00677" name="l00677"></a><span class="lineno">  677</span>            }</div>
<div class="line"><a id="l00678" name="l00678"></a><span class="lineno">  678</span>            <span class="keywordflow">if</span> (line.find(<span class="stringliteral">&quot;\&quot;type\&quot;: &quot;</span>) != std::string::npos &amp;&amp; reading_node) {</div>
<div class="line"><a id="l00679" name="l00679"></a><span class="lineno">  679</span>                std::string pattern = <span class="stringliteral">&quot;\&quot;type\&quot;: &quot;</span>;</div>
<div class="line"><a id="l00680" name="l00680"></a><span class="lineno">  680</span>                <span class="keywordtype">size_t</span> startPos = line.find(pattern);</div>
<div class="line"><a id="l00681" name="l00681"></a><span class="lineno">  681</span>                startPos += pattern.length();</div>
<div class="line"><a id="l00682" name="l00682"></a><span class="lineno">  682</span>                <span class="keyword">const</span> <span class="keywordtype">size_t</span> firstQuote = line.find(<span class="charliteral">&#39;\&quot;&#39;</span>, startPos);</div>
<div class="line"><a id="l00683" name="l00683"></a><span class="lineno">  683</span>                <span class="keyword">const</span> <span class="keywordtype">size_t</span> secondQuote = line.find(<span class="charliteral">&#39;\&quot;&#39;</span>, firstQuote + 1);</div>
<div class="line"><a id="l00684" name="l00684"></a><span class="lineno">  684</span>                type = line.substr(firstQuote + 1, secondQuote - firstQuote - 1);</div>
<div class="line"><a id="l00685" name="l00685"></a><span class="lineno">  685</span>                <span class="keywordflow">continue</span>;</div>
<div class="line"><a id="l00686" name="l00686"></a><span class="lineno">  686</span>            }</div>
<div class="line"><a id="l00687" name="l00687"></a><span class="lineno">  687</span>            <span class="keywordflow">if</span> (line.find(<span class="stringliteral">&quot;\&quot;name\&quot;: &quot;</span>) != std::string::npos &amp;&amp; reading_node) {</div>
<div class="line"><a id="l00688" name="l00688"></a><span class="lineno">  688</span>                std::string pattern = <span class="stringliteral">&quot;\&quot;name\&quot;: &quot;</span>;</div>
<div class="line"><a id="l00689" name="l00689"></a><span class="lineno">  689</span>                <span class="keywordtype">size_t</span> startPos = line.find(pattern);</div>
<div class="line"><a id="l00690" name="l00690"></a><span class="lineno">  690</span>                startPos += pattern.length();</div>
<div class="line"><a id="l00691" name="l00691"></a><span class="lineno">  691</span>                <span class="keyword">const</span> <span class="keywordtype">size_t</span> firstQuote = line.find(<span class="charliteral">&#39;\&quot;&#39;</span>, startPos);</div>
<div class="line"><a id="l00692" name="l00692"></a><span class="lineno">  692</span>                <span class="keyword">const</span> <span class="keywordtype">size_t</span> secondQuote = line.find(<span class="charliteral">&#39;\&quot;&#39;</span>, firstQuote + 1);</div>
<div class="line"><a id="l00693" name="l00693"></a><span class="lineno">  693</span>                name = line.substr(firstQuote + 1, secondQuote - firstQuote - 1);</div>
<div class="line"><a id="l00694" name="l00694"></a><span class="lineno">  694</span>                <span class="keywordflow">continue</span>;</div>
<div class="line"><a id="l00695" name="l00695"></a><span class="lineno">  695</span>            }</div>
<div class="line"><a id="l00696" name="l00696"></a><span class="lineno">  696</span>            <span class="keywordflow">if</span> (line.find(<span class="stringliteral">&quot;\&quot;pre\&quot;: &quot;</span>) != std::string::npos &amp;&amp; reading_node) {</div>
<div class="line"><a id="l00697" name="l00697"></a><span class="lineno">  697</span>                pre.clear();</div>
<div class="line"><a id="l00698" name="l00698"></a><span class="lineno">  698</span>                std::string pattern = <span class="stringliteral">&quot;\&quot;pre\&quot;: &quot;</span>;</div>
<div class="line"><a id="l00699" name="l00699"></a><span class="lineno">  699</span>                <span class="keywordtype">size_t</span> startPos = line.find(pattern);</div>
<div class="line"><a id="l00700" name="l00700"></a><span class="lineno">  700</span>                startPos += pattern.length();</div>
<div class="line"><a id="l00701" name="l00701"></a><span class="lineno">  701</span>                <span class="keyword">const</span> <span class="keywordtype">size_t</span> firstQuote = line.find(<span class="charliteral">&#39;[&#39;</span>, startPos);</div>
<div class="line"><a id="l00702" name="l00702"></a><span class="lineno">  702</span>                <span class="keyword">const</span> <span class="keywordtype">size_t</span> secondQuote = line.find(<span class="charliteral">&#39;]&#39;</span>, firstQuote + 1);</div>
<div class="line"><a id="l00703" name="l00703"></a><span class="lineno">  703</span>                std::string pre_str = line.substr(firstQuote + 1, secondQuote - firstQuote - 1);</div>
<div class="line"><a id="l00704" name="l00704"></a><span class="lineno">  704</span>                std::stringstream ss(pre_str);</div>
<div class="line"><a id="l00705" name="l00705"></a><span class="lineno">  705</span>                <span class="keywordtype">int</span> val;</div>
<div class="line"><a id="l00706" name="l00706"></a><span class="lineno">  706</span>                <span class="keywordflow">while</span> (ss &gt;&gt; val) {</div>
<div class="line"><a id="l00707" name="l00707"></a><span class="lineno">  707</span>                    pre.push_back(val);</div>
<div class="line"><a id="l00708" name="l00708"></a><span class="lineno">  708</span>                    <span class="keywordflow">if</span> (ss.peek() == <span class="charliteral">&#39;,&#39;</span>) {</div>
<div class="line"><a id="l00709" name="l00709"></a><span class="lineno">  709</span>                        ss.ignore();</div>
<div class="line"><a id="l00710" name="l00710"></a><span class="lineno">  710</span>                    }</div>
<div class="line"><a id="l00711" name="l00711"></a><span class="lineno">  711</span>                }</div>
<div class="line"><a id="l00712" name="l00712"></a><span class="lineno">  712</span>                <span class="keywordflow">continue</span>;</div>
<div class="line"><a id="l00713" name="l00713"></a><span class="lineno">  713</span>            }</div>
<div class="line"><a id="l00714" name="l00714"></a><span class="lineno">  714</span>            <span class="keywordflow">if</span> (line.find(<span class="stringliteral">&quot;\&quot;post\&quot;: &quot;</span>) != std::string::npos &amp;&amp; reading_node) {</div>
<div class="line"><a id="l00715" name="l00715"></a><span class="lineno">  715</span>                post.clear();</div>
<div class="line"><a id="l00716" name="l00716"></a><span class="lineno">  716</span>                std::string pattern = <span class="stringliteral">&quot;\&quot;post\&quot;: &quot;</span>;</div>
<div class="line"><a id="l00717" name="l00717"></a><span class="lineno">  717</span>                <span class="keywordtype">size_t</span> startPos = line.find(pattern);</div>
<div class="line"><a id="l00718" name="l00718"></a><span class="lineno">  718</span>                startPos += pattern.length();</div>
<div class="line"><a id="l00719" name="l00719"></a><span class="lineno">  719</span>                <span class="keyword">const</span> <span class="keywordtype">size_t</span> firstQuote = line.find(<span class="charliteral">&#39;[&#39;</span>, startPos);</div>
<div class="line"><a id="l00720" name="l00720"></a><span class="lineno">  720</span>                <span class="keyword">const</span> <span class="keywordtype">size_t</span> secondQuote = line.find(<span class="charliteral">&#39;]&#39;</span>, firstQuote + 1);</div>
<div class="line"><a id="l00721" name="l00721"></a><span class="lineno">  721</span>                std::string post_str = line.substr(firstQuote + 1, secondQuote - firstQuote - 1);</div>
<div class="line"><a id="l00722" name="l00722"></a><span class="lineno">  722</span>                std::stringstream ss(post_str);</div>
<div class="line"><a id="l00723" name="l00723"></a><span class="lineno">  723</span>                <span class="keywordtype">int</span> val;</div>
<div class="line"><a id="l00724" name="l00724"></a><span class="lineno">  724</span>                <span class="keywordflow">while</span> (ss &gt;&gt; val) {</div>
<div class="line"><a id="l00725" name="l00725"></a><span class="lineno">  725</span>                    post.push_back(val);</div>
<div class="line"><a id="l00726" name="l00726"></a><span class="lineno">  726</span>                    <span class="keywordflow">if</span> (ss.peek() == <span class="charliteral">&#39;,&#39;</span>) {</div>
<div class="line"><a id="l00727" name="l00727"></a><span class="lineno">  727</span>                        ss.ignore();</div>
<div class="line"><a id="l00728" name="l00728"></a><span class="lineno">  728</span>                    }</div>
<div class="line"><a id="l00729" name="l00729"></a><span class="lineno">  729</span>                }</div>
<div class="line"><a id="l00730" name="l00730"></a><span class="lineno">  730</span>                <span class="keywordflow">continue</span>;</div>
<div class="line"><a id="l00731" name="l00731"></a><span class="lineno">  731</span>            }</div>
<div class="line"><a id="l00732" name="l00732"></a><span class="lineno">  732</span>            <span class="keywordflow">if</span> (line.find(<span class="stringliteral">&quot;\&quot;shape\&quot;: &quot;</span>) != std::string::npos &amp;&amp; reading_node) {</div>
<div class="line"><a id="l00733" name="l00733"></a><span class="lineno">  733</span>                std::string pattern = <span class="stringliteral">&quot;\&quot;shape\&quot;: &quot;</span>;</div>
<div class="line"><a id="l00734" name="l00734"></a><span class="lineno">  734</span>                <span class="keywordtype">size_t</span> startPos = line.find(pattern);</div>
<div class="line"><a id="l00735" name="l00735"></a><span class="lineno">  735</span>                startPos += pattern.length();</div>
<div class="line"><a id="l00736" name="l00736"></a><span class="lineno">  736</span>                <span class="keyword">const</span> <span class="keywordtype">size_t</span> firstQuote = line.find(<span class="charliteral">&#39;[&#39;</span>, startPos);</div>
<div class="line"><a id="l00737" name="l00737"></a><span class="lineno">  737</span>                <span class="keyword">const</span> <span class="keywordtype">size_t</span> secondQuote = line.find(<span class="charliteral">&#39;]&#39;</span>, firstQuote + 1);</div>
<div class="line"><a id="l00738" name="l00738"></a><span class="lineno">  738</span>                std::string shape_str = line.substr(firstQuote + 1, secondQuote - firstQuote - 1);</div>
<div class="line"><a id="l00739" name="l00739"></a><span class="lineno">  739</span>                std::stringstream ss(shape_str);</div>
<div class="line"><a id="l00740" name="l00740"></a><span class="lineno">  740</span>                <span class="keywordtype">int</span> val;</div>
<div class="line"><a id="l00741" name="l00741"></a><span class="lineno">  741</span>                <span class="keyword">auto</span> index = 0;</div>
<div class="line"><a id="l00742" name="l00742"></a><span class="lineno">  742</span>                <span class="keywordflow">while</span> (ss &gt;&gt; val) {</div>
<div class="line"><a id="l00743" name="l00743"></a><span class="lineno">  743</span>                    shape[index] = val;</div>
<div class="line"><a id="l00744" name="l00744"></a><span class="lineno">  744</span>                    index++;</div>
<div class="line"><a id="l00745" name="l00745"></a><span class="lineno">  745</span>                    <span class="keywordflow">if</span> (ss.peek() == <span class="charliteral">&#39;,&#39;</span>) {</div>
<div class="line"><a id="l00746" name="l00746"></a><span class="lineno">  746</span>                        ss.ignore();</div>
<div class="line"><a id="l00747" name="l00747"></a><span class="lineno">  747</span>                    }</div>
<div class="line"><a id="l00748" name="l00748"></a><span class="lineno">  748</span>                }</div>
<div class="line"><a id="l00749" name="l00749"></a><span class="lineno">  749</span>                <span class="keywordflow">continue</span>;</div>
<div class="line"><a id="l00750" name="l00750"></a><span class="lineno">  750</span>            }</div>
<div class="line"><a id="l00751" name="l00751"></a><span class="lineno">  751</span>            <span class="keywordflow">if</span> (line.find(<span class="stringliteral">&quot;\&quot;data\&quot;: &quot;</span>) != std::string::npos &amp;&amp; reading_node) {</div>
<div class="line"><a id="l00752" name="l00752"></a><span class="lineno">  752</span>                data = <span class="keyword">static_cast&lt;</span><span class="keywordtype">float</span>*<span class="keyword">&gt;</span>(malloc(<span class="keyword">sizeof</span>(<span class="keywordtype">float</span>) * shape[0] * shape[1]));</div>
<div class="line"><a id="l00753" name="l00753"></a><span class="lineno">  753</span>                std::string pattern = <span class="stringliteral">&quot;\&quot;data\&quot;: &quot;</span>;</div>
<div class="line"><a id="l00754" name="l00754"></a><span class="lineno">  754</span>                <span class="keywordtype">size_t</span> startPos = line.find(pattern);</div>
<div class="line"><a id="l00755" name="l00755"></a><span class="lineno">  755</span>                startPos += pattern.length();</div>
<div class="line"><a id="l00756" name="l00756"></a><span class="lineno">  756</span>                <span class="keyword">const</span> <span class="keywordtype">size_t</span> firstQuote = line.find(<span class="charliteral">&#39;[&#39;</span>, startPos);</div>
<div class="line"><a id="l00757" name="l00757"></a><span class="lineno">  757</span>                <span class="keyword">const</span> <span class="keywordtype">size_t</span> secondQuote = line.find(<span class="charliteral">&#39;]&#39;</span>, firstQuote + 1);</div>
<div class="line"><a id="l00758" name="l00758"></a><span class="lineno">  758</span>                std::string data_str = line.substr(firstQuote + 1, secondQuote - firstQuote - 1);</div>
<div class="line"><a id="l00759" name="l00759"></a><span class="lineno">  759</span>                std::stringstream ss(data_str);</div>
<div class="line"><a id="l00760" name="l00760"></a><span class="lineno">  760</span>                <span class="keywordtype">float</span> val;</div>
<div class="line"><a id="l00761" name="l00761"></a><span class="lineno">  761</span>                i = 0;</div>
<div class="line"><a id="l00762" name="l00762"></a><span class="lineno">  762</span>                <span class="keywordflow">while</span> (ss &gt;&gt; val) {</div>
<div class="line"><a id="l00763" name="l00763"></a><span class="lineno">  763</span>                    data[i++] = val;</div>
<div class="line"><a id="l00764" name="l00764"></a><span class="lineno">  764</span>                    <span class="keywordflow">if</span> (ss.peek() == <span class="charliteral">&#39;,&#39;</span>) {</div>
<div class="line"><a id="l00765" name="l00765"></a><span class="lineno">  765</span>                        ss.ignore();</div>
<div class="line"><a id="l00766" name="l00766"></a><span class="lineno">  766</span>                    }</div>
<div class="line"><a id="l00767" name="l00767"></a><span class="lineno">  767</span>                }</div>
<div class="line"><a id="l00768" name="l00768"></a><span class="lineno">  768</span>                <span class="keywordflow">continue</span>;</div>
<div class="line"><a id="l00769" name="l00769"></a><span class="lineno">  769</span>            }</div>
<div class="line"><a id="l00770" name="l00770"></a><span class="lineno">  770</span>            <span class="keywordflow">if</span> (line.find(<span class="stringliteral">&quot;\&quot;requires_grad\&quot;: &quot;</span>) != std::string::npos &amp;&amp; reading_node) {</div>
<div class="line"><a id="l00771" name="l00771"></a><span class="lineno">  771</span>                std::string pattern = <span class="stringliteral">&quot;\&quot;requires_grad\&quot;: &quot;</span>;</div>
<div class="line"><a id="l00772" name="l00772"></a><span class="lineno">  772</span>                <span class="keywordtype">size_t</span> startPos = line.find(pattern);</div>
<div class="line"><a id="l00773" name="l00773"></a><span class="lineno">  773</span>                startPos += pattern.length();</div>
<div class="line"><a id="l00774" name="l00774"></a><span class="lineno">  774</span>                <span class="keywordflow">if</span> (line.substr(startPos, 4) == <span class="stringliteral">&quot;true&quot;</span>) {</div>
<div class="line"><a id="l00775" name="l00775"></a><span class="lineno">  775</span>                    requires_grad = <span class="keyword">true</span>;</div>
<div class="line"><a id="l00776" name="l00776"></a><span class="lineno">  776</span>                }</div>
<div class="line"><a id="l00777" name="l00777"></a><span class="lineno">  777</span>                <span class="keywordflow">else</span> {</div>
<div class="line"><a id="l00778" name="l00778"></a><span class="lineno">  778</span>                    requires_grad = <span class="keyword">false</span>;</div>
<div class="line"><a id="l00779" name="l00779"></a><span class="lineno">  779</span>                }</div>
<div class="line"><a id="l00780" name="l00780"></a><span class="lineno">  780</span>                <span class="keywordflow">continue</span>;</div>
<div class="line"><a id="l00781" name="l00781"></a><span class="lineno">  781</span>            }</div>
<div class="line"><a id="l00782" name="l00782"></a><span class="lineno">  782</span>            <span class="keywordflow">if</span> (line.find(<span class="stringliteral">&quot;\&quot;grad\&quot;: &quot;</span>) != std::string::npos &amp;&amp; reading_node &amp;&amp; requires_grad) {</div>
<div class="line"><a id="l00783" name="l00783"></a><span class="lineno">  783</span>                grad = <span class="keyword">static_cast&lt;</span><span class="keywordtype">float</span>*<span class="keyword">&gt;</span>(malloc(<span class="keyword">sizeof</span>(<span class="keywordtype">float</span>) * shape[0] * shape[1]));</div>
<div class="line"><a id="l00784" name="l00784"></a><span class="lineno">  784</span>                std::string pattern = <span class="stringliteral">&quot;\&quot;grad\&quot;: &quot;</span>;</div>
<div class="line"><a id="l00785" name="l00785"></a><span class="lineno">  785</span>                <span class="keywordtype">size_t</span> startPos = line.find(pattern);</div>
<div class="line"><a id="l00786" name="l00786"></a><span class="lineno">  786</span>                startPos += pattern.length();</div>
<div class="line"><a id="l00787" name="l00787"></a><span class="lineno">  787</span>                <span class="keyword">const</span> <span class="keywordtype">size_t</span> firstQuote = line.find(<span class="charliteral">&#39;[&#39;</span>, startPos);</div>
<div class="line"><a id="l00788" name="l00788"></a><span class="lineno">  788</span>                <span class="keyword">const</span> <span class="keywordtype">size_t</span> secondQuote = line.find(<span class="charliteral">&#39;]&#39;</span>, firstQuote + 1);</div>
<div class="line"><a id="l00789" name="l00789"></a><span class="lineno">  789</span>                std::string grad_str = line.substr(firstQuote + 1, secondQuote - firstQuote - 1);</div>
<div class="line"><a id="l00790" name="l00790"></a><span class="lineno">  790</span>                std::stringstream ss(grad_str);</div>
<div class="line"><a id="l00791" name="l00791"></a><span class="lineno">  791</span>                <span class="keywordtype">float</span> val;</div>
<div class="line"><a id="l00792" name="l00792"></a><span class="lineno">  792</span>                i = 0;</div>
<div class="line"><a id="l00793" name="l00793"></a><span class="lineno">  793</span>                <span class="keywordflow">while</span> (ss &gt;&gt; val) {</div>
<div class="line"><a id="l00794" name="l00794"></a><span class="lineno">  794</span>                    grad[i++] = val;</div>
<div class="line"><a id="l00795" name="l00795"></a><span class="lineno">  795</span>                    <span class="keywordflow">if</span> (ss.peek() == <span class="charliteral">&#39;,&#39;</span>) {</div>
<div class="line"><a id="l00796" name="l00796"></a><span class="lineno">  796</span>                        ss.ignore();</div>
<div class="line"><a id="l00797" name="l00797"></a><span class="lineno">  797</span>                    }</div>
<div class="line"><a id="l00798" name="l00798"></a><span class="lineno">  798</span>                }</div>
<div class="line"><a id="l00799" name="l00799"></a><span class="lineno">  799</span>                <span class="keywordflow">continue</span>;</div>
<div class="line"><a id="l00800" name="l00800"></a><span class="lineno">  800</span>            }</div>
<div class="line"><a id="l00801" name="l00801"></a><span class="lineno">  801</span>            <span class="keywordflow">if</span> (line.find(<span class="charliteral">&#39;}&#39;</span>) != std::string::npos &amp;&amp; reading_node) {</div>
<div class="line"><a id="l00802" name="l00802"></a><span class="lineno">  802</span>                reading_node = <span class="keyword">false</span>;</div>
<div class="line"><a id="l00803" name="l00803"></a><span class="lineno">  803</span>                <a class="code hl_friend" href="classnz_1_1graph_1_1_compute_graph.html#a11f567efba2c857e64a88c411c5e6c54">CreateNode</a>(<span class="keyword">this</span>, type, name, pre, shape, data, requires_grad, grad);</div>
<div class="line"><a id="l00804" name="l00804"></a><span class="lineno">  804</span>                free(data);</div>
<div class="line"><a id="l00805" name="l00805"></a><span class="lineno">  805</span>                <span class="keywordflow">if</span> (requires_grad) {</div>
<div class="line"><a id="l00806" name="l00806"></a><span class="lineno">  806</span>                    free(grad);</div>
<div class="line"><a id="l00807" name="l00807"></a><span class="lineno">  807</span>                }</div>
<div class="line"><a id="l00808" name="l00808"></a><span class="lineno">  808</span>                <span class="keywordflow">continue</span>;</div>
<div class="line"><a id="l00809" name="l00809"></a><span class="lineno">  809</span>            }</div>
<div class="line"><a id="l00810" name="l00810"></a><span class="lineno">  810</span>        }</div>
<div class="line"><a id="l00811" name="l00811"></a><span class="lineno">  811</span>    }</div>
</div>
<div class="line"><a id="l00812" name="l00812"></a><span class="lineno">  812</span> </div>
<div class="foldopen" id="foldopen00813" data-start="{" data-end="}">
<div class="line"><a id="l00813" name="l00813"></a><span class="lineno"><a class="line" href="classnz_1_1graph_1_1_compute_graph.html#a5ae0d765ed40c4973b54dc20a5cddf61">  813</a></span>    <a class="code hl_class" href="classnz_1_1nodes_1_1_node.html">Node</a>* <a class="code hl_function" href="classnz_1_1graph_1_1_compute_graph.html#a5ae0d765ed40c4973b54dc20a5cddf61">ComputeGraph::operator[]</a>(<span class="keyword">const</span> std::string&amp; name) {</div>
<div class="line"><a id="l00814" name="l00814"></a><span class="lineno">  814</span>        <span class="keywordflow">return</span> nodeRoster[name];</div>
<div class="line"><a id="l00815" name="l00815"></a><span class="lineno">  815</span>    }</div>
</div>
<div class="line"><a id="l00816" name="l00816"></a><span class="lineno">  816</span> </div>
<div class="foldopen" id="foldopen00817" data-start="{" data-end="}">
<div class="line"><a id="l00817" name="l00817"></a><span class="lineno"><a class="line" href="classnz_1_1graph_1_1_compute_graph.html#a1e5d98f57514c6f7a539de2776f48c5d">  817</a></span>    std::string <a class="code hl_function" href="classnz_1_1graph_1_1_compute_graph.html#a1e5d98f57514c6f7a539de2776f48c5d">ComputeGraph::nodesList</a>() {</div>
<div class="line"><a id="l00818" name="l00818"></a><span class="lineno">  818</span>        <span class="keywordtype">size_t</span> name_width = 4;</div>
<div class="line"><a id="l00819" name="l00819"></a><span class="lineno">  819</span>        <span class="keywordtype">size_t</span> type_width = 4;</div>
<div class="line"><a id="l00820" name="l00820"></a><span class="lineno">  820</span> </div>
<div class="line"><a id="l00821" name="l00821"></a><span class="lineno">  821</span>        <span class="keywordflow">for</span> (<span class="keyword">const</span> <span class="keyword">auto</span>&amp; [name, node] : nodeRoster) {</div>
<div class="line"><a id="l00822" name="l00822"></a><span class="lineno">  822</span>            <span class="keywordflow">if</span> (name.length() &gt; name_width) name_width = name.length();</div>
<div class="line"><a id="l00823" name="l00823"></a><span class="lineno">  823</span>            <span class="keywordflow">if</span> (node-&gt;type.length() &gt; type_width) type_width = node-&gt;type.length();</div>
<div class="line"><a id="l00824" name="l00824"></a><span class="lineno">  824</span>        }</div>
<div class="line"><a id="l00825" name="l00825"></a><span class="lineno">  825</span> </div>
<div class="line"><a id="l00826" name="l00826"></a><span class="lineno">  826</span>        std::ostringstream oss;</div>
<div class="line"><a id="l00827" name="l00827"></a><span class="lineno">  827</span>        oss &lt;&lt; std::left;</div>
<div class="line"><a id="l00828" name="l00828"></a><span class="lineno">  828</span> </div>
<div class="line"><a id="l00829" name="l00829"></a><span class="lineno">  829</span>        oss &lt;&lt; std::setw(static_cast&lt;long long&gt;(name_width)) &lt;&lt; <span class="stringliteral">&quot;Name&quot;</span> &lt;&lt; <span class="stringliteral">&quot; | &quot;</span></div>
<div class="line"><a id="l00830" name="l00830"></a><span class="lineno">  830</span>            &lt;&lt; std::setw(<span class="keyword">static_cast&lt;</span><span class="keywordtype">long</span> <span class="keywordtype">long</span><span class="keyword">&gt;</span>(type_width)) &lt;&lt; <span class="stringliteral">&quot;Type&quot;</span>;</div>
<div class="line"><a id="l00831" name="l00831"></a><span class="lineno">  831</span> </div>
<div class="line"><a id="l00832" name="l00832"></a><span class="lineno">  832</span>        <span class="keywordflow">for</span> (<span class="keyword">const</span> <span class="keyword">auto</span>&amp; [name, node] : nodeRoster) {</div>
<div class="line"><a id="l00833" name="l00833"></a><span class="lineno">  833</span>            oss &lt;&lt; <span class="stringliteral">&quot;\n&quot;</span></div>
<div class="line"><a id="l00834" name="l00834"></a><span class="lineno">  834</span>                &lt;&lt; std::setw(<span class="keyword">static_cast&lt;</span><span class="keywordtype">long</span> <span class="keywordtype">long</span><span class="keyword">&gt;</span>(name_width)) &lt;&lt; name &lt;&lt; <span class="stringliteral">&quot; | &quot;</span></div>
<div class="line"><a id="l00835" name="l00835"></a><span class="lineno">  835</span>                &lt;&lt; std::setw(<span class="keyword">static_cast&lt;</span><span class="keywordtype">long</span> <span class="keywordtype">long</span><span class="keyword">&gt;</span>(type_width)) &lt;&lt; node-&gt;type;</div>
<div class="line"><a id="l00836" name="l00836"></a><span class="lineno">  836</span>        }</div>
<div class="line"><a id="l00837" name="l00837"></a><span class="lineno">  837</span> </div>
<div class="line"><a id="l00838" name="l00838"></a><span class="lineno">  838</span>        <span class="keywordflow">return</span> oss.str();</div>
<div class="line"><a id="l00839" name="l00839"></a><span class="lineno">  839</span>    }</div>
</div>
<div class="line"><a id="l00840" name="l00840"></a><span class="lineno">  840</span>} <span class="comment">// Graph</span></div>
<div class="line"><a id="l00841" name="l00841"></a><span class="lineno">  841</span><span class="comment">// nz</span></div>
<div class="ttc" id="a_compute_graph_8cuh_html"><div class="ttname"><a href="_compute_graph_8cuh.html">ComputeGraph.cuh</a></div><div class="ttdoc">Defines the ComputeGraph class for constructing and managing computational graphs in neural network m...</div></div>
<div class="ttc" id="aclassnz_1_1cu_strm_1_1_stream_manager_html_ab4b2eb422e0e1ee44bdfdc0eb94457ce"><div class="ttname"><a href="classnz_1_1cu_strm_1_1_stream_manager.html#ab4b2eb422e0e1ee44bdfdc0eb94457ce">nz::cuStrm::StreamManager::Instance</a></div><div class="ttdeci">static StreamManager &amp; Instance()</div><div class="ttdoc">Returns a reference to the singleton instance of the StreamManager.</div><div class="ttdef"><b>Definition</b> <a href="_stream_manager_8cuh_source.html#l00154">StreamManager.cuh:154</a></div></div>
<div class="ttc" id="aclassnz_1_1cu_strm_1_1_stream_manager_html_abe439fa00c0bd369c0b2345b095ed5af"><div class="ttname"><a href="classnz_1_1cu_strm_1_1_stream_manager.html#abe439fa00c0bd369c0b2345b095ed5af">nz::cuStrm::StreamManager::syncData</a></div><div class="ttdeci">void syncData(T *data)</div><div class="ttdoc">Synchronizes host thread with completion events for a specific data object.</div><div class="ttdef"><b>Definition</b> <a href="_stream_manager_8cuh_source.html#l00714">StreamManager.cuh:714</a></div></div>
<div class="ttc" id="aclassnz_1_1cu_strm_1_1_stream_manager_html_afa38d5c6db0e6b48c8f74ce8ad0df2bc"><div class="ttname"><a href="classnz_1_1cu_strm_1_1_stream_manager.html#afa38d5c6db0e6b48c8f74ce8ad0df2bc">nz::cuStrm::StreamManager::memcpy</a></div><div class="ttdeci">void memcpy(T *dst, T *src, const size_t size, const cudaMemcpyKind kind)</div><div class="ttdoc">Asynchronously copies data between CUDA device and host memory based on the specified memory copy kin...</div><div class="ttdef"><b>Definition</b> <a href="_stream_manager_8cuh_source.html#l00391">StreamManager.cuh:391</a></div></div>
<div class="ttc" id="aclassnz_1_1data_1_1_dimension_html"><div class="ttname"><a href="classnz_1_1data_1_1_dimension.html">nz::data::Dimension</a></div><div class="ttdoc">Represents a multi - dimensional shape, typically used in deep learning for tensor dimensions.</div><div class="ttdef"><b>Definition</b> <a href="_dimension_8cuh_source.html#l00057">Dimension.cuh:57</a></div></div>
<div class="ttc" id="aclassnz_1_1data_1_1_tensor_html"><div class="ttname"><a href="classnz_1_1data_1_1_tensor.html">nz::data::Tensor</a></div><div class="ttdoc">A class for representing and manipulating multidimensional arrays (tensors) in GPU memory.</div><div class="ttdef"><b>Definition</b> <a href="_tensor_8cuh_source.html#l00134">Tensor.cuh:134</a></div></div>
<div class="ttc" id="aclassnz_1_1graph_1_1_compute_graph_html"><div class="ttname"><a href="classnz_1_1graph_1_1_compute_graph.html">nz::graph::ComputeGraph</a></div><div class="ttdoc">Represents a computational graph, which manages nodes and the computation flow.</div><div class="ttdef"><b>Definition</b> <a href="_compute_graph_8cuh_source.html#l00224">ComputeGraph.cuh:224</a></div></div>
<div class="ttc" id="aclassnz_1_1graph_1_1_compute_graph_html_a0c5a596f38e6d7e8c1c78e19b9b56167"><div class="ttname"><a href="classnz_1_1graph_1_1_compute_graph.html#a0c5a596f38e6d7e8c1c78e19b9b56167">nz::graph::ComputeGraph::forward</a></div><div class="ttdeci">void forward()</div><div class="ttdoc">Performs forward propagation on the computational graph.</div><div class="ttdef"><b>Definition</b> <a href="#l00405">ComputeGraph.cu:405</a></div></div>
<div class="ttc" id="aclassnz_1_1graph_1_1_compute_graph_html_a11f567efba2c857e64a88c411c5e6c54"><div class="ttname"><a href="classnz_1_1graph_1_1_compute_graph.html#a11f567efba2c857e64a88c411c5e6c54">nz::graph::ComputeGraph::CreateNode</a></div><div class="ttdeci">friend DL_API void CreateNode(ComputeGraph *graph, const std::string &amp;type, const std::string &amp;name, std::vector&lt; int &gt; pre, const Tensor::shape_type &amp;shape, float *data, bool requires_grad, float *grad)</div><div class="ttdoc">Creates and adds a node to the computational graph based on the specified type.</div><div class="ttdef"><b>Definition</b> <a href="#l00108">ComputeGraph.cu:108</a></div></div>
<div class="ttc" id="aclassnz_1_1graph_1_1_compute_graph_html_a1d54fe96c53fec967c8a20e1ac79ceed"><div class="ttname"><a href="classnz_1_1graph_1_1_compute_graph.html#a1d54fe96c53fec967c8a20e1ac79ceed">nz::graph::ComputeGraph::print</a></div><div class="ttdeci">std::ostream &amp; print(std::ostream &amp;os)</div><div class="ttdoc">Prints the details of the computational graph to the provided output stream.</div><div class="ttdef"><b>Definition</b> <a href="#l00008">ComputeGraph.cu:8</a></div></div>
<div class="ttc" id="aclassnz_1_1graph_1_1_compute_graph_html_a1e5d98f57514c6f7a539de2776f48c5d"><div class="ttname"><a href="classnz_1_1graph_1_1_compute_graph.html#a1e5d98f57514c6f7a539de2776f48c5d">nz::graph::ComputeGraph::nodesList</a></div><div class="ttdeci">std::string nodesList()</div><div class="ttdoc">Generates a formatted string representing the list of nodes in the ComputeGraph.</div><div class="ttdef"><b>Definition</b> <a href="#l00817">ComputeGraph.cu:817</a></div></div>
<div class="ttc" id="aclassnz_1_1graph_1_1_compute_graph_html_a3a53a15c04e85da4e6aa0a209fbee4c2"><div class="ttname"><a href="classnz_1_1graph_1_1_compute_graph.html#a3a53a15c04e85da4e6aa0a209fbee4c2">nz::graph::ComputeGraph::getLoss</a></div><div class="ttdeci">Tensor::value_type getLoss() const</div><div class="ttdoc">Retrieves the loss value from the first output node in the computational graph.</div><div class="ttdef"><b>Definition</b> <a href="#l00534">ComputeGraph.cu:534</a></div></div>
<div class="ttc" id="aclassnz_1_1graph_1_1_compute_graph_html_a3f94019acc205a47726c0b0797b0e631"><div class="ttname"><a href="classnz_1_1graph_1_1_compute_graph.html#a3f94019acc205a47726c0b0797b0e631">nz::graph::ComputeGraph::zeroGrad</a></div><div class="ttdeci">void zeroGrad() const</div><div class="ttdoc">Resets the gradients of all nodes in the computational graph.</div><div class="ttdef"><b>Definition</b> <a href="#l00434">ComputeGraph.cu:434</a></div></div>
<div class="ttc" id="aclassnz_1_1graph_1_1_compute_graph_html_a47cde0bdd9c961d55bfd16c9adb8810e"><div class="ttname"><a href="classnz_1_1graph_1_1_compute_graph.html#a47cde0bdd9c961d55bfd16c9adb8810e">nz::graph::ComputeGraph::addOutput</a></div><div class="ttdeci">OutputNode * addOutput(OutputNode *node, const std::string &amp;name=&quot;default&quot;)</div><div class="ttdoc">Adds an output node to the computational graph.</div><div class="ttdef"><b>Definition</b> <a href="#l00389">ComputeGraph.cu:389</a></div></div>
<div class="ttc" id="aclassnz_1_1graph_1_1_compute_graph_html_a5960e75d631200994c0a2f78f58674dd"><div class="ttname"><a href="classnz_1_1graph_1_1_compute_graph.html#a5960e75d631200994c0a2f78f58674dd">nz::graph::ComputeGraph::topologicalSort</a></div><div class="ttdeci">void topologicalSort()</div><div class="ttdoc">Performs topological sorting on the computational graph.</div><div class="ttdef"><b>Definition</b> <a href="#l00258">ComputeGraph.cu:258</a></div></div>
<div class="ttc" id="aclassnz_1_1graph_1_1_compute_graph_html_a5ae0d765ed40c4973b54dc20a5cddf61"><div class="ttname"><a href="classnz_1_1graph_1_1_compute_graph.html#a5ae0d765ed40c4973b54dc20a5cddf61">nz::graph::ComputeGraph::operator[]</a></div><div class="ttdeci">Node * operator[](const std::string &amp;name)</div><div class="ttdoc">Retrieves the node associated with the given name in the computational graph.</div><div class="ttdef"><b>Definition</b> <a href="#l00813">ComputeGraph.cu:813</a></div></div>
<div class="ttc" id="aclassnz_1_1graph_1_1_compute_graph_html_a5ce9a271dcea6cc2b9490cd815ee53ed"><div class="ttname"><a href="classnz_1_1graph_1_1_compute_graph.html#a5ce9a271dcea6cc2b9490cd815ee53ed">nz::graph::ComputeGraph::addNode</a></div><div class="ttdeci">NodeType * addNode(NodeType *node, const std::string &amp;name=&quot;default&quot;)</div><div class="ttdoc">Adds a node of any type to the computational graph.</div><div class="ttdef"><b>Definition</b> <a href="_compute_graph_8cuh_source.html#l00511">ComputeGraph.cuh:511</a></div></div>
<div class="ttc" id="aclassnz_1_1graph_1_1_compute_graph_html_a69d293299714e489ce8cc2dba1e5a9a5"><div class="ttname"><a href="classnz_1_1graph_1_1_compute_graph.html#a69d293299714e489ce8cc2dba1e5a9a5">nz::graph::ComputeGraph::getOutputNode</a></div><div class="ttdeci">OutputNode * getOutputNode() const</div><div class="ttdoc">Retrieves the first output node in the computational graph.</div><div class="ttdef"><b>Definition</b> <a href="#l00527">ComputeGraph.cu:527</a></div></div>
<div class="ttc" id="aclassnz_1_1graph_1_1_compute_graph_html_a6aa07603223ea714d74fc884218cb50f"><div class="ttname"><a href="classnz_1_1graph_1_1_compute_graph.html#a6aa07603223ea714d74fc884218cb50f">nz::graph::ComputeGraph::save</a></div><div class="ttdeci">void save(const std::string &amp;path)</div><div class="ttdoc">Saves the current computational graph to a JSON file.</div><div class="ttdef"><b>Definition</b> <a href="#l00553">ComputeGraph.cu:553</a></div></div>
<div class="ttc" id="aclassnz_1_1graph_1_1_compute_graph_html_a75a4b72c93448c8617a7af41cc471897"><div class="ttname"><a href="classnz_1_1graph_1_1_compute_graph.html#a75a4b72c93448c8617a7af41cc471897">nz::graph::ComputeGraph::addInput</a></div><div class="ttdeci">InputNode * addInput(const Tensor::shape_type &amp;shape, bool requires_grad=false, const std::string &amp;name=&quot;default&quot;)</div><div class="ttdoc">Adds an input node to the computational graph.</div><div class="ttdef"><b>Definition</b> <a href="#l00302">ComputeGraph.cu:302</a></div></div>
<div class="ttc" id="aclassnz_1_1graph_1_1_compute_graph_html_a75f918fb756d5f7155ee49092ce713e1"><div class="ttname"><a href="classnz_1_1graph_1_1_compute_graph.html#a75f918fb756d5f7155ee49092ce713e1">nz::graph::ComputeGraph::getOutput</a></div><div class="ttdeci">Tensor::value_type * getOutput() const</div><div class="ttdoc">Retrieves the output data of the first output node in the computational graph.</div><div class="ttdef"><b>Definition</b> <a href="#l00508">ComputeGraph.cu:508</a></div></div>
<div class="ttc" id="aclassnz_1_1graph_1_1_compute_graph_html_a856ffb66fa42357ab1a12f9f74d2aba4"><div class="ttname"><a href="classnz_1_1graph_1_1_compute_graph.html#a856ffb66fa42357ab1a12f9f74d2aba4">nz::graph::ComputeGraph::randomize</a></div><div class="ttdeci">void randomize(const std::string &amp;name, unsigned long long seed=0)</div><div class="ttdoc">Randomizes the output tensor of a specified node in the computational graph.</div><div class="ttdef"><b>Definition</b> <a href="#l00440">ComputeGraph.cu:440</a></div></div>
<div class="ttc" id="aclassnz_1_1graph_1_1_compute_graph_html_a8b3498c50429d631b07b5906e7455614"><div class="ttname"><a href="classnz_1_1graph_1_1_compute_graph.html#a8b3498c50429d631b07b5906e7455614">nz::graph::ComputeGraph::setInput</a></div><div class="ttdeci">void setInput(const std::string &amp;name, Tensor::value_type *data)</div><div class="ttdoc">Sets the input data for a specified node in the computational graph.</div><div class="ttdef"><b>Definition</b> <a href="#l00490">ComputeGraph.cu:490</a></div></div>
<div class="ttc" id="aclassnz_1_1graph_1_1_compute_graph_html_a90e4737107c4ec96a1a95022daa6de2f"><div class="ttname"><a href="classnz_1_1graph_1_1_compute_graph.html#a90e4737107c4ec96a1a95022daa6de2f">nz::graph::ComputeGraph::randomizeAll</a></div><div class="ttdeci">void randomizeAll() const</div><div class="ttdoc">Randomizes the output tensors of all input nodes in the computational graph.</div><div class="ttdef"><b>Definition</b> <a href="#l00458">ComputeGraph.cu:458</a></div></div>
<div class="ttc" id="aclassnz_1_1graph_1_1_compute_graph_html_a9748d859d952864bfa498a79c9fb394a"><div class="ttname"><a href="classnz_1_1graph_1_1_compute_graph.html#a9748d859d952864bfa498a79c9fb394a">nz::graph::ComputeGraph::load</a></div><div class="ttdeci">void load(const std::string &amp;path)</div><div class="ttdoc">Loads a computational graph from a JSON file.</div><div class="ttdef"><b>Definition</b> <a href="#l00642">ComputeGraph.cu:642</a></div></div>
<div class="ttc" id="aclassnz_1_1graph_1_1_compute_graph_html_adf21c05c4515016d34d5fc7d639b9e40"><div class="ttname"><a href="classnz_1_1graph_1_1_compute_graph.html#adf21c05c4515016d34d5fc7d639b9e40">nz::graph::ComputeGraph::fillAll</a></div><div class="ttdeci">void fillAll(Tensor::value_type val) const</div><div class="ttdoc">Fills the output tensors of all input nodes with a given value.</div><div class="ttdef"><b>Definition</b> <a href="#l00484">ComputeGraph.cu:484</a></div></div>
<div class="ttc" id="aclassnz_1_1graph_1_1_compute_graph_html_adfcad402fe85084ddbb4f8b6415adc05"><div class="ttname"><a href="classnz_1_1graph_1_1_compute_graph.html#adfcad402fe85084ddbb4f8b6415adc05">nz::graph::ComputeGraph::backward</a></div><div class="ttdeci">void backward()</div><div class="ttdoc">Performs backward propagation on the computational graph.</div><div class="ttdef"><b>Definition</b> <a href="#l00414">ComputeGraph.cu:414</a></div></div>
<div class="ttc" id="aclassnz_1_1graph_1_1_compute_graph_html_ae732145241d1da547219a3e8b95a04ee"><div class="ttname"><a href="classnz_1_1graph_1_1_compute_graph.html#ae732145241d1da547219a3e8b95a04ee">nz::graph::ComputeGraph::fill</a></div><div class="ttdeci">void fill(const std::string &amp;name, Tensor::value_type val)</div><div class="ttdoc">Fills the output tensor of a specified node with a given value.</div><div class="ttdef"><b>Definition</b> <a href="#l00466">ComputeGraph.cu:466</a></div></div>
<div class="ttc" id="aclassnz_1_1graph_1_1_compute_graph_html_aede9d05b6c8e7394ea730b5ffea42164"><div class="ttname"><a href="classnz_1_1graph_1_1_compute_graph.html#aede9d05b6c8e7394ea730b5ffea42164">nz::graph::ComputeGraph::isSorted</a></div><div class="ttdeci">bool isSorted() const</div><div class="ttdoc">Checks whether the computational graph has been topologically sorted.</div><div class="ttdef"><b>Definition</b> <a href="#l00298">ComputeGraph.cu:298</a></div></div>
<div class="ttc" id="aclassnz_1_1graph_1_1_compute_graph_html_aee306895a3078a0b09d2a4bdf4843200"><div class="ttname"><a href="classnz_1_1graph_1_1_compute_graph.html#aee306895a3078a0b09d2a4bdf4843200">nz::graph::ComputeGraph::update</a></div><div class="ttdeci">void update(Optimizer *optimizer) const</div><div class="ttdoc">Updates the parameters of the nodes that require gradients using the provided optimizer.</div><div class="ttdef"><b>Definition</b> <a href="#l00541">ComputeGraph.cu:541</a></div></div>
<div class="ttc" id="aclassnz_1_1graph_1_1_compute_graph_html_aef02d3290fb92b142be346f2fdb125c5"><div class="ttname"><a href="classnz_1_1graph_1_1_compute_graph.html#aef02d3290fb92b142be346f2fdb125c5">nz::graph::ComputeGraph::getOutputHost</a></div><div class="ttdeci">Tensor::value_type * getOutputHost() const</div><div class="ttdoc">Retrieves the output data of the first output node in the computational graph and copies it to host m...</div><div class="ttdef"><b>Definition</b> <a href="#l00515">ComputeGraph.cu:515</a></div></div>
<div class="ttc" id="aclassnz_1_1nodes_1_1_node_html"><div class="ttname"><a href="classnz_1_1nodes_1_1_node.html">nz::nodes::Node</a></div><div class="ttdoc">Base class for nodes in a neural network or computational graph.</div><div class="ttdef"><b>Definition</b> <a href="_nodes_8cuh_source.html#l00114">Nodes.cuh:114</a></div></div>
<div class="ttc" id="aclassnz_1_1nodes_1_1calc_1_1_add_node_html"><div class="ttname"><a href="classnz_1_1nodes_1_1calc_1_1_add_node.html">nz::nodes::calc::AddNode</a></div><div class="ttdoc">Represents a node that performs element-wise addition between two input tensors.</div><div class="ttdef"><b>Definition</b> <a href="_nodes_8cuh_source.html#l00917">Nodes.cuh:917</a></div></div>
<div class="ttc" id="aclassnz_1_1nodes_1_1calc_1_1_e_l_u_node_html"><div class="ttname"><a href="classnz_1_1nodes_1_1calc_1_1_e_l_u_node.html">nz::nodes::calc::ELUNode</a></div><div class="ttdoc">Represents an Exponential Linear Unit (ELU) activation function node in a computational graph.</div><div class="ttdef"><b>Definition</b> <a href="_nodes_8cuh_source.html#l02648">Nodes.cuh:2648</a></div></div>
<div class="ttc" id="aclassnz_1_1nodes_1_1calc_1_1_hard_sigmoid_node_html"><div class="ttname"><a href="classnz_1_1nodes_1_1calc_1_1_hard_sigmoid_node.html">nz::nodes::calc::HardSigmoidNode</a></div><div class="ttdoc">Represents a Hard Sigmoid activation function node in a computational graph.</div><div class="ttdef"><b>Definition</b> <a href="_nodes_8cuh_source.html#l02803">Nodes.cuh:2803</a></div></div>
<div class="ttc" id="aclassnz_1_1nodes_1_1calc_1_1_hard_swish_node_html"><div class="ttname"><a href="classnz_1_1nodes_1_1calc_1_1_hard_swish_node.html">nz::nodes::calc::HardSwishNode</a></div><div class="ttdoc">Represents a Hard Swish activation function node in a computational graph.</div><div class="ttdef"><b>Definition</b> <a href="_nodes_8cuh_source.html#l02954">Nodes.cuh:2954</a></div></div>
<div class="ttc" id="aclassnz_1_1nodes_1_1calc_1_1_leaky_re_l_u_node_html"><div class="ttname"><a href="classnz_1_1nodes_1_1calc_1_1_leaky_re_l_u_node.html">nz::nodes::calc::LeakyReLUNode</a></div><div class="ttdoc">Represents a Leaky Rectified Linear Unit (LeakyReLU) activation function node in a computational grap...</div><div class="ttdef"><b>Definition</b> <a href="_nodes_8cuh_source.html#l02353">Nodes.cuh:2353</a></div></div>
<div class="ttc" id="aclassnz_1_1nodes_1_1calc_1_1_mat_mul_node_html"><div class="ttname"><a href="classnz_1_1nodes_1_1calc_1_1_mat_mul_node.html">nz::nodes::calc::MatMulNode</a></div><div class="ttdoc">Represents a matrix multiplication operation node in a computational graph.</div><div class="ttdef"><b>Definition</b> <a href="_nodes_8cuh_source.html#l01060">Nodes.cuh:1060</a></div></div>
<div class="ttc" id="aclassnz_1_1nodes_1_1calc_1_1_re_l_u_node_html"><div class="ttname"><a href="classnz_1_1nodes_1_1calc_1_1_re_l_u_node.html">nz::nodes::calc::ReLUNode</a></div><div class="ttdoc">Represents a Rectified Linear Unit (ReLU) operation node in a computational graph.</div><div class="ttdef"><b>Definition</b> <a href="_nodes_8cuh_source.html#l01929">Nodes.cuh:1929</a></div></div>
<div class="ttc" id="aclassnz_1_1nodes_1_1calc_1_1_sigmoid_node_html"><div class="ttname"><a href="classnz_1_1nodes_1_1calc_1_1_sigmoid_node.html">nz::nodes::calc::SigmoidNode</a></div><div class="ttdoc">Represents a Sigmoid activation function node in a computational graph.</div><div class="ttdef"><b>Definition</b> <a href="_nodes_8cuh_source.html#l02072">Nodes.cuh:2072</a></div></div>
<div class="ttc" id="aclassnz_1_1nodes_1_1calc_1_1_softmax_node_html"><div class="ttname"><a href="classnz_1_1nodes_1_1calc_1_1_softmax_node.html">nz::nodes::calc::SoftmaxNode</a></div><div class="ttdoc">Implements the Softmax activation function as a node in a neural network computational graph.</div><div class="ttdef"><b>Definition</b> <a href="_nodes_8cuh_source.html#l03152">Nodes.cuh:3152</a></div></div>
<div class="ttc" id="aclassnz_1_1nodes_1_1calc_1_1_sub_node_html"><div class="ttname"><a href="classnz_1_1nodes_1_1calc_1_1_sub_node.html">nz::nodes::calc::SubNode</a></div><div class="ttdoc">Represents a subtraction operation node in a computational graph.</div><div class="ttdef"><b>Definition</b> <a href="_nodes_8cuh_source.html#l01787">Nodes.cuh:1787</a></div></div>
<div class="ttc" id="aclassnz_1_1nodes_1_1calc_1_1_swish_node_html"><div class="ttname"><a href="classnz_1_1nodes_1_1calc_1_1_swish_node.html">nz::nodes::calc::SwishNode</a></div><div class="ttdoc">Represents a Swish activation function node in a computational graph.</div><div class="ttdef"><b>Definition</b> <a href="_nodes_8cuh_source.html#l02504">Nodes.cuh:2504</a></div></div>
<div class="ttc" id="aclassnz_1_1nodes_1_1calc_1_1_tanh_node_html"><div class="ttname"><a href="classnz_1_1nodes_1_1calc_1_1_tanh_node.html">nz::nodes::calc::TanhNode</a></div><div class="ttdoc">Represents a hyperbolic tangent (tanh) activation function node in a computational graph.</div><div class="ttdef"><b>Definition</b> <a href="_nodes_8cuh_source.html#l02214">Nodes.cuh:2214</a></div></div>
<div class="ttc" id="aclassnz_1_1nodes_1_1io_1_1_input_node_html"><div class="ttname"><a href="classnz_1_1nodes_1_1io_1_1_input_node.html">nz::nodes::io::InputNode</a></div><div class="ttdoc">Represents an input node in a computational graph.</div><div class="ttdef"><b>Definition</b> <a href="_nodes_8cuh_source.html#l00437">Nodes.cuh:437</a></div></div>
<div class="ttc" id="aclassnz_1_1nodes_1_1io_1_1_output_node_html"><div class="ttname"><a href="classnz_1_1nodes_1_1io_1_1_output_node.html">nz::nodes::io::OutputNode</a></div><div class="ttdoc">Base class for loss function nodes in a computational graph.</div><div class="ttdef"><b>Definition</b> <a href="_nodes_8cuh_source.html#l00683">Nodes.cuh:683</a></div></div>
<div class="ttc" id="aclassnz_1_1nodes_1_1loss_1_1_binary_cross_entropy_node_html"><div class="ttname"><a href="classnz_1_1nodes_1_1loss_1_1_binary_cross_entropy_node.html">nz::nodes::loss::BinaryCrossEntropyNode</a></div><div class="ttdoc">Represents the Binary Cross-Entropy (BCE) loss function node in a computational graph.</div><div class="ttdef"><b>Definition</b> <a href="_nodes_8cuh_source.html#l04958">Nodes.cuh:4958</a></div></div>
<div class="ttc" id="aclassnz_1_1nodes_1_1loss_1_1_mean_squared_error_node_html"><div class="ttname"><a href="classnz_1_1nodes_1_1loss_1_1_mean_squared_error_node.html">nz::nodes::loss::MeanSquaredErrorNode</a></div><div class="ttdoc">Represents the Mean Squared Error (MSE) loss function node in a computational graph.</div><div class="ttdef"><b>Definition</b> <a href="_nodes_8cuh_source.html#l04804">Nodes.cuh:4804</a></div></div>
<div class="ttc" id="aclassnz_1_1opt_1_1_optimizer_html"><div class="ttname"><a href="classnz_1_1opt_1_1_optimizer.html">nz::opt::Optimizer</a></div><div class="ttdoc">Base class for optimization algorithms in deep learning.</div><div class="ttdef"><b>Definition</b> <a href="_optimizer_8cuh_source.html#l00125">Optimizer.cuh:125</a></div></div>
<div class="ttc" id="aclassnz_1_1opt_1_1_optimizer_html_a826381abaaf29dbebade7cfd38b266e4"><div class="ttname"><a href="classnz_1_1opt_1_1_optimizer.html#a826381abaaf29dbebade7cfd38b266e4">nz::opt::Optimizer::step</a></div><div class="ttdeci">virtual void step(Node *input)=0</div><div class="ttdoc">Pure virtual function for performing a single optimization step.</div></div>
<div class="ttc" id="anamespacenz_1_1graph_html"><div class="ttname"><a href="namespacenz_1_1graph.html">nz::graph</a></div><div class="ttdoc">Contains classes and functions for managing and executing computation graphs in deep learning workflo...</div></div>
<div class="ttc" id="anamespacenz_1_1graph_html_a3c6187d695022b05df227d913ac89d91"><div class="ttname"><a href="namespacenz_1_1graph.html#a3c6187d695022b05df227d913ac89d91">nz::graph::CreateNode</a></div><div class="ttdeci">void CreateNode(ComputeGraph *graph, const std::string &amp;type, const std::string &amp;name, std::vector&lt; int &gt; pre, const Tensor::shape_type &amp;shape, float *data, const bool requires_grad, float *grad)</div><div class="ttdoc">Creates and adds a node to the computational graph based on the specified type.</div><div class="ttdef"><b>Definition</b> <a href="#l00108">ComputeGraph.cu:108</a></div></div>
<div class="ttc" id="anamespacenz_1_1nodes_html_a42672c2d7708ae1d0c071fd9bef6c03c"><div class="ttname"><a href="namespacenz_1_1nodes.html#a42672c2d7708ae1d0c071fd9bef6c03c">nz::nodes::operator&lt;&lt;</a></div><div class="ttdeci">std::enable_if_t&lt; std::is_base_of_v&lt; Node, T &gt;, std::ostream &amp; &gt; operator&lt;&lt;(std::ostream &amp;os, const T &amp;node)</div><div class="ttdoc">Overloads the &lt;&lt; operator to print information about a node.</div><div class="ttdef"><b>Definition</b> <a href="_nodes_8cuh_source.html#l00114">Nodes.cuh:352</a></div></div>
</div><!-- fragment --></div><!-- contents -->
<!-- start footer part -->
<hr class="footer"/><address class="footer"><small>
Generated by&#160;<a href="https://www.doxygen.org/index.html"><img class="footer" src="doxygen.svg" width="104" height="31" alt="doxygen"/></a> 1.12.0
</small></address>
</div><!-- doc-content -->
</body>
</html>
